DCLGAN
DCLGAN copied to clipboard
How to resume a training
Hello Junlin,
Thank you very much for your code!
I was just wondering, what is the command to resume training from a checkpoint?
I tried this one, but it didn't work out:
python3 train.py --dataroot ./datasets/NIR2RGB --name NIR2RGB_DCL --pretrained_name NIR2RGB_DCL
Thank you.
Hi Victor, Thanks for your nice workds!
For resume training, you might add --continue_train --epoch_count N, where N is the current epoch of the pretrained model. continue_train will load your latest checkpoint from the checkpoint path.
So I guess you can try:
python3 train.py --dataroot ./datasets/NIR2RGB --name NIR2RGB_DCL --continue_train --epoch_count N
Hope it helps.
Thanks, it works!