Skip to content

Commit c8b4511

Browse files
committed
A bit of an optimizer overhaul, added an improved factory, list_optimizers, class helper and add info classes with descriptions, arg configs
1 parent e6d72ed commit c8b4511

19 files changed

+1127
-770
lines changed

tests/test_optim.py

Lines changed: 12 additions & 296 deletions
Large diffs are not rendered by default.

timm/optim/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,5 @@
1717
from .rmsprop_tf import RMSpropTF
1818
from .sgdp import SGDP
1919

20-
from .optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs
20+
from ._optim_factory import list_optimizers, get_optimizer_class, create_optimizer_v2, \
21+
create_optimizer, optimizer_kwargs, OptimInfo, OptimizerRegistry

0 commit comments

Comments
 (0)