A bit of an optimizer overhaul, added an improved factory, list_optim…

…izers, class helper and add info classes with descriptions, arg configs
huggingface · Nov 13, 2024 · c8b4511 · c8b4511
1 parent e6d72ed
commit c8b4511
Show file tree

Hide file tree

Showing 19 changed files with 1,127 additions and 770 deletions.
diff --git a/tests/test_optim.py b/tests/test_optim.py
diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py
@@ -17,4 +17,5 @@
 from .rmsprop_tf import RMSpropTF
 from .sgdp import SGDP
 
-from .optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs
+from ._optim_factory import list_optimizers, get_optimizer_class, create_optimizer_v2, \
+    create_optimizer, optimizer_kwargs, OptimInfo, OptimizerRegistry