CycleGAN icon indicating copy to clipboard operation
CycleGAN copied to clipboard

How do i resume my training on colab?

Open kartikJ-9 opened this issue 4 years ago • 1 comments

Decent results require 3k-5k frames. My GPU session on colab gets disconnected due to usage while training. I am saving the checkpoints in the drive. Is there any way I can resume the training from a particular epoch? I have a sequence of images obtained from a video.

kartikJ-9 avatar Apr 16 '20 07:04 kartikJ-9

I assume that you are using the Pytorch version with colab. For Lua, you can use continue_train flag. For PyTorch, please see this Q & A.

junyanz avatar Apr 16 '20 19:04 junyanz