search

jettify / pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch
Apache License 2.0
3k stars 294 forks source link

Checking for requires_grad #518

Open gedomada opened 1 year ago

gedomada commented 1 year ago

Added validation for the requires_grad attribute. Necessary for training with modules containing Int Parameters.