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

Fixed StepLR_Restart learning scheme #3

Closed victorca25 closed 5 years ago

victorca25 commented 5 years ago

Updated for the fixed StepLR_Restart learning scheme, now it's consistent with the training strategy published in the PPON paper and additionally, can be trained using MultiStepLR_Restart.

These options are configurable in the JSON file, which now contain both options pre-configured to try to replicate the paper variables.