how to save epoches every 20 times and load model again? i want to train this model on colab and it has 1000 epoches ..but its takes too long.
