How to set the number of training steps?

[Saury997]

In your article, you stated that the model was trained in 10,000 steps, however, I was unable to find the variable that set the number of training steps from the code's config file. Can you explain how to set this parameter? Thank you so much!

[lin-tianyu]

This is because there is no such variable that sets the training steps haha.

In practice, I used tensorboard to monitor the training process. You can stop the training manually using kill -9 $PID when the metrics converge.

tensorboard in our code can also monitor losses and the generated images to give you comprehensive information about the training process. Highly recommended.