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

[Feature Request] Curriculum Training for Augmentations #52

Open Kim2091 opened 3 years ago

Kim2091 commented 3 years ago

The ability to change which augmentation preset is being used at different points in training would be great. For example, at 10k iterations, resrgan_blur could be used, but at 30k it's automatically switched to bsrgan_blur.

This was discussed in the #trainner channel on the GU Discord server

Edit: A possible expansion on this idea, augmentation preset strengths. I'm not sure how it'd function, but I figured I'd bring it up