search

dansuh17 / segan-pytorch

SEGAN pytorch implementation https://arxiv.org/abs/1703.09452
GNU General Public License v3.0
106 stars 32 forks source link

Save the Optimizer #14

Closed ahmed-fau closed 6 years ago

ahmed-fau commented 6 years ago

Hi @deNsuh ,

It seems that resuming the training process is incomplete as the optimizers for both generator and discriminator networks are not being saved.

Best Regards

dansuh17 commented 6 years ago

True. Saving the state of the optimizer will not be difficult. Are you willing to do a pull-request with this? If not, I'll be happy to implement this myself :)

ahmed-fau commented 6 years ago

Unfortunately I am still not proficient with Github utilities and edit the scripts locally :( Sorry for that and thanks for your support.

dansuh17 commented 6 years ago

Sorry for the late work. Saving the optimizer states as well as model states are now implemented.