search

MichiganCOG / ViP

Video Platform for Action Recognition and Object Detection in Pytorch
MIT License
219 stars 37 forks source link

Pytorch definition of optimizers and schedulers #52

Open ehofesmann opened 4 years ago

ehofesmann commented 4 years ago

Allow you to specify the exact pytorch class and related parameters in the yaml file for the optimizer and scheduler you want to run. (e.g. torch.schedulers.MultiStepLR)