Skip to content

Commit ba85080

Browse files
Yuan-Jinghuirwightman
authored andcommitted
Add cadamp optimizer to optimization factory
1 parent a3643d2 commit ba85080

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

timm/optim/_optim_factory.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -780,6 +780,13 @@ def _register_cautious_optimizers(registry: OptimizerRegistry) -> None:
780780
has_momentum=True,
781781
defaults={'nesterov': True, 'caution': True}
782782
),
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+
),
783790
]
784791
for opt in cautious_optimizers:
785792
registry.register(opt)

0 commit comments

Comments
 (0)