Нужны предложения Tensorflow - PullRequest
0 голосов
/ 27 апреля 2018

У меня есть набор данных 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".

Может ли кто-нибудь сказать мне, сколько шагов я должен выбрать для точных прогнозов?
Я новичок в глубоком обучении, и предложения будут полезны, поскольку я нахожусь в фазе обучения.

1 Ответ

0 голосов
/ 27 апреля 2018

Если у вас есть около 3000 изображений на букву и 26 букв, это дает вам около 78 000 изображений на эпоху . Если размер вашей партии составляет b , то это дает вам 78 000 / b тренировочных шагов за эпоху. Я бы посоветовал сначала потренироваться до 10 эпох и посмотреть, что получится.

Это экспериментальная наука. Выведите точность после каждой эпохи и посмотрите, что произойдет, если сеть улучшится. Прекратите тренировку, когда она прекратит улучшаться.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...