Solved by ptrblck in post #4 
              
              
                That would be probably a better idea! 
Similar to the last epoch number in case you would like to continue the training in another script.