Pytorch definition of optimizers and schedulers #52
Description
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)
{{ message }}
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)