torch_optimizer has no attribute 'SGD'
I am trying to import torch_optimizer on Google Colab. I have successfully !pip installed torch_optimizer and then imported it. However, every attribute I call with torch_optimizer gives an attribute error:
AttributeError: module 'torch_optimizer' has no attribute 'SGD'
This holds true for SGD, Adam, etc.
Here is a photo of my code. Thanks!
Pytorch Optimizer

I have checked the documentation of pytorch-optimizer. The vanilla SGD is not there. There are small modifications of SGD as AccSGD, SGDW, SGDP etc. You can use the simple pytorch optimizer torch.optim.SGD. Check this visualization script where they are comparing the baseline SGD to other methods implemented by this library. Same thing holds for Adam as well. Pretty clear from the script.
