search

neurostatslab / vocalocator

Deep neural networks for sound source localization and vocalization attribution.
MIT License
2 stars 0 forks source link

Allow user to specify learning rate schedule in model config #27

Closed amanchoudhri closed 1 year ago

amanchoudhri commented 1 year ago

Would be great to include support for all the schedulers in torch.optim.lr_scheduler but at the very least, we should have:

Maybe in a config subdict 'LR_SCHEDULE_PARAMS' under 'OPTIMIZATION'?