search

openclimatefix / satflow

Satellite Optical Flow with machine learning models
https://satflow.readthedocs.io/en/stable/
MIT License
61 stars 10 forks source link

Add LR warm up scheduling #36

Closed jacobbieker closed 3 years ago

jacobbieker commented 3 years ago

Based off https://twitter.com/jack_kelly/status/1411615500371640320?s=19 might help with metnet training, should be easy to add and experiment with. Plan would be to add it to the generic model interface so we can try it on any of the models

jacobbieker commented 3 years ago

Added for MetNet in #48