How to stops training after specied number of steps?
derekdrAi opened this issue · 1 comments
derekdrAi commented
I want to stop training after a specified number of steps automatically, does anyone knows how to do it?
apurvaasf commented
By default, the number of epochs is set to indefinite, so it will keep on training till you kill the process.
You can kill the process and use the last checkpoint file.
Hope this helps!