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.

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	More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.	3 years ago
adahessian.py	Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.	4 years ago
adamp.py	More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.	3 years ago
adamw.py	…
lamb.py	…
lars.py	fix lars	3 years ago
lookahead.py	…
madgrad.py	…
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	More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.	3 years ago
sgdp.py	…