Как узнать общее количество шагов во время обучения с использованием Tensorflow Object Detection API?

Question

Последние 3 часа я выполнял учебное задание на облачной машине с графическим процессором со следующей командой:

python legacy/train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

, и после этого в журнале об этом говорится, например:

 INFO:tensorflow:global step 14455: loss = 0.5896 (0.775 sec/step)
I1001 19:27:43.575182 140054916601600 tf_logging.py:116] global step 14455: loss = 0.5896 (0.775 sec/step)

Как узнать, сколько шагов нужно сделать или сколько всего шагов?

Answer 1

В ssd_mobilenet_v1_pets.config в строке указано 163 :

num_steps: 200000

Это общее количество шагов, которое будет выполнено обучающим сценарием, если вы не внесли никаких изменений.

Answer 2

Если вы используете предварительно определенную топологию модели, вы просматриваете период обучения (в эпохах или шагах) в документации, поставляемой с моделью.Если вы сделали свою собственную модель, вы определяете период обучения, наблюдая за результатами теста.Когда точность достигает приемлемого уровня, а затем начинает падать, вы, вероятно, перетренируетесь.Вернитесь к высокой точке точности.Повторите этот эксперимент несколько раз, чтобы определить «сладкое пятно» для вашей модели.