Path: tree/master/labml_nn/optimizers
4910 views
Name | Size | Last Modified |
|---|---|---|
| __init__.py | 8.1 KB | |
| ada_belief.py | 6.8 KB | |
| adam.py | 8.6 KB | |
| adam_fp16.py | 5.5 KB | |
| adam_warmup.py | 2.3 KB | |
| adam_warmup_cosine_decay.py | 3.6 KB | |
| amsgrad.py | 7.8 KB | |
| configs.py | 5.1 KB | |
| mnist_experiment.py | 3.7 KB | |
| noam.py | 3.3 KB | |
| performance_test.py | 1.6 KB | |
| radam.py | 11.2 KB | |
| readme.md | 568 bytes | |
| sophia.py | 7.2 KB |