victorca25 / traiNNer

traiNNer: Deep learning framework for image and video super-resolution, restoration and image-to-image translation, for training and testing.
Apache License 2.0
293 stars 39 forks source link

lr not updated upon niter change #30

Open keywae opened 3 years ago

keywae commented 3 years ago

I've noticed that if you update niter mid training, it displays the correct new 'lr steps', but it does not correct the current 'lr rate', in accordance with the new steps. Looks like, it just keep the old lr rate from 'latest.state' file.

Example: When i change niter from 100,000 to 500,000, following were the logs:

Herer is the settings in config file: lr_steps_rel: [0.1, 0.2, 0.4, 0.6] lr_G: 0.0001