You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/optim
Ross Wightman f35d6ea57b
Add multi-tensor (foreach) version of Lion in style of upcoming PyTorch 2.0 optimizers
2 years ago
..
__init__.py
adabelief.py
adafactor.py
adahessian.py
adamp.py
adamw.py
adan.py
lamb.py
lars.py
lion.py Add multi-tensor (foreach) version of Lion in style of upcoming PyTorch 2.0 optimizers 2 years ago
lookahead.py
madgrad.py
nadam.py
nvnovograd.py
optim_factory.py Add multi-tensor (foreach) version of Lion in style of upcoming PyTorch 2.0 optimizers 2 years ago
radam.py
rmsprop_tf.py
sgdp.py