diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:22:13 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:22:13 +0100 |
| commit | 69e93847bb0adf6e819f48e70e6a88da4cf0b7f6 (patch) | |
| tree | b7e533674d51f8e34a9318b7a4f7344aae56f5de /training/conf/optimizers/radam.yaml | |
| parent | 68640a70a769568fd571e50322d6e3fa40c78271 (diff) | |
fix: nuke madgrad, add radam
Diffstat (limited to 'training/conf/optimizers/radam.yaml')
| -rw-r--r-- | training/conf/optimizers/radam.yaml | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/training/conf/optimizers/radam.yaml b/training/conf/optimizers/radam.yaml new file mode 100644 index 0000000..7ee1234 --- /dev/null +++ b/training/conf/optimizers/radam.yaml @@ -0,0 +1,7 @@ +radam: + _target_: torch.optim.RAdam + lr: 1.5e-4 + betas: [0.9, 0.999] + weight_decay: 1.0e-4 + eps: 1.0e-8 + parameters: network |