Я следую этому уроку https://www.tensorflow.org/hub/tutorials/image_retraining, чтобы обучить свой собственный набор данных с использованием предварительно обученной модели Inception V3.
В retrain.py (полный код: https://raw.githubusercontent.com/tensorflow/hub/master/examples/image_retraining/retrain.py) Я не нахожу опцию эпохи, я нашел только training_steps
parser.add_argument(
'--how_many_training_steps',
type=int,
default=4000,
help='How many training steps to run before ending.'
)
Из этого В чем разница между шагами и эпохами в TensorFlow? , я обнаружил, что training_steps - это то же самое, что и итерация . Итак, как я могу указать, сколько эпох я хочу установить? Спасибо.