The output doesn’t necessarily indicate that the training is restarting, as the loss value seems to be stuck at ~6, so you might want to play around with some hyperparameters of your training, such as the learning rate etc.