时间戳

首页 > 解决方案 > 需要 TensorFlow 建议

android - 需要 TensorFlow 建议

问题描述

我有一个ASL（美国手语）数据集，其中每个字母有 3000 张图像，我将通过tensorflow codelabs使用这个脚本来训练我的模型

 "python -m scripts.retrain \ 
         --bottleneck_dir=tf_files/bottlenecks \
         --how_many_training_steps=? \ 
         --model_dir=tf_files/models/ \ --summaries_dir=tf_files/training_summaries/"mobilenet_1.0_224" \ 
         --output_graph=tf_files/retrained_graph.pb \ 
         --output_labels=tf_files/retrained_labels.txt \ 
         --architecture="mobilenet_1.0_224" \ --image_dir=tf_files/dataset".

谁能告诉我必须选择多少步才能进行准确的预测？
我是深度学习的新手，因为我处于学习阶段，所以建议会很有帮助。

标签： androidtensorflow

解决方案

如果每个字母有大约 3,000 个图像，并且有 26 个字母，那么每个epoch大约有 78,000 个图像。如果您的批量大小为b，那么每个 epoch 为您提供 78,000/ b个训练步骤。我建议先训练到 10 个 epoch，然后看看会发生什么。

这是实验科学。在每个 epoch 之后打印准确度，看看会发生什么，如果网络进一步改进。当训练停止显着改善时停止训练。

推荐阅读