Can i resume training if Colab disconnects while training from the last successful epoch..

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

Other

22.76k stars 6.29k forks source link

Can i resume training if Colab disconnects while training from the last successful epoch.. #1133

Open shivom9713 opened 4 years ago

shivom9713 commented 4 years ago

SIr , my dataset has 1500 image pairs in pix2pix i have trained the model for 4 hours and it crashed after that . Even i couldnt fetch results from it as i did not save the checkpoint when the server disconnected . Can you please help me in the same .

yajain commented 3 years ago

You can use the option --continue_train. Also set --epoch_count to specify a different starting epoch count.