We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a3643d2 commit ba85080Copy full SHA for ba85080
1 file changed
timm/optim/_optim_factory.py
@@ -780,6 +780,13 @@ def _register_cautious_optimizers(registry: OptimizerRegistry) -> None:
780
has_momentum=True,
781
defaults={'nesterov': True, 'caution': True}
782
),
783
+ OptimInfo(
784
+ name='cadamp',
785
+ opt_class=AdamP,
786
+ description='Add the spherical cautious optimizer and the standard cautious optimizer to AdamP',
787
+ has_betas=True,
788
+ defaults={'wd_ratio': 0.01, 'nesterov': True, 'caution': True}
789
+ ),
790
]
791
for opt in cautious_optimizers:
792
registry.register(opt)
0 commit comments