我正在按照本教程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.'
)
在此What is the difference between steps and epochs in TensorFlow?中,我发现 training_steps 与迭代相同。那么,如何指定要设置的纪元?谢谢。