BMSG-GAN icon indicating copy to clipboard operation
BMSG-GAN copied to clipboard
verified publisher icon akanimax

I want to stop and resume the training continuously.

Open bemoregt opened this issue 5 years ago • 3 comments

Hi, @markstrefford @owang @sridharmahadevan @akanimax @huangzh13

I want to stop and resume the training continuously with saved model.

My resuming commad : "$ python3 train.py --start 300" # I stop at 350 epoch

But, My Training sample is started from the scratch. --;

What's wrong to me ??

Thanks. Best, @bemoregt.

bemoregt avatar Mar 13 '20 08:03 bemoregt

I am also curious about this. The start flag does not seem to resume the training, only store with another number for the files... It would be great to have a resume flag.

materialvision avatar Apr 14 '20 16:04 materialvision

Also want to be able to continue from a given point. I tried just now, after training for 100 epoch, starting at 101, it appears to have started from scratch again.

ademus4 avatar Jul 05 '21 15:07 ademus4

Ah figured this out; you can set the path to the previous weights (generator and discriminator), using the input parameters in the train.py file, lines 35 to 51

ademus4 avatar Jul 07 '21 11:07 ademus4