PyTorch-StudioGAN icon indicating copy to clipboard operation
PyTorch-StudioGAN copied to clipboard

How to resume training from checkpoint weights

Open gaurav104 opened this issue 3 years ago • 1 comments

How can I resume the GAN I have been training, if I canceled the training, using the weights being saved in the checkpoints folder?

Thanks

gaurav104 avatar Apr 26 '22 00:04 gaurav104

You can use the exact command that you've used when training with a single addition of -ckpt LOCATION_TO_YOUR_CHECKPOINT_FOLDER option. The model will then train from the last saved checkpoint. If you specify -best option, training will continue from the best checkpoint so far.

alex4727 avatar Apr 30 '22 09:04 alex4727