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 0557c8257d
Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc.
3 years ago
..
__init__.py Add lamb/lars to optim init imports, remove stray comment 3 years ago
adabelief.py
adafactor.py
adahessian.py
adamp.py
adamw.py
lamb.py MOAR optimizer changes. Woo! 3 years ago
lars.py fix lars 3 years ago
lookahead.py
madgrad.py MOAR optimizer changes. Woo! 3 years ago
nadam.py
nvnovograd.py
optim_factory.py Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc. 3 years ago
radam.py
rmsprop_tf.py
sgdp.py