Adaptive lr to training.py

balbasty / nitorch

Neuroimaging in PyTorch

Other

86 stars 14 forks source link

Closed brudfors closed 3 years ago

brudfors commented 3 years ago

Had to create a new PR because of mixing up two branches..

I have added your comments @balbasty, as ReduceLROnPlateau is the only sched that takes an input, maybe we can leave just using isinstance() for now?

balbasty commented 3 years ago

I am fine with that. You can merge ;)