search

jettify / pytorch-optimizer

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

`AttributeError: module 'torch_optimizer' has no attribute 'MAGRAD'` #537

Open SamMohel opened 7 months ago

SamMohel commented 7 months ago

I tired to follow the example of using MAGRAD optimizer 

  import torch_optimizer as optim
   optimizer = optim.MAGRAD(model.parameters(), lr=0.1)
   optimizer.zero_grad()
   loss_fn(model(input), target).backward()
   optimizer.step()

but got AttributeError: module 'torch_optimizer' has no attribute 'MAGRAD'