Как отобразить количество эпох в API обнаружения тензорного потока с помощью Faster Rcnn?

Question

Я использую API обнаружения объектов Tensorflow. Что я понял, прочитав файл fast_rcnn_inception_v2_pets.config , так это то, что num_steps означает общее количество шагов, а не эпох. Но тогда какой смысл указывать batch_size ?? Допустим, у меня есть 500 изображений в моих тренировочных данных, и я установил размер партии = 5 и num_steps = 20k. Значит ли это, что количество эпох равно 200 ??

Когда я запускаю model_main.py, он показывает только потерю global_steps. Поэтому, если эти глобальные шаги не являются эпохами, тогда как мне изменить код, чтобы отображать потерю поезда и потерю стоимости после каждого шага, а также после каждой эпохи.

Answer 1

Итак, вы правы, предполагая, что у вас 200 эпох.

У меня была похожая проблема с отсутствием потери.Мое решение состояло в том, чтобы перейти к файлу model_main.py и затем вставить

tf.logging.set_verbosity (tf.logging.INFO)

после импорта.затем он показывает вам потери после каждых 100 шагов.Вы можете изменить set_verbosity, если хотите иметь его после каждой эпохи;)