Post merge fix reference of old param groups helper fn locations

huggingface · Nov 13, 2024 · 5dae918 · 5dae918
1 parent 326e5dc
commit 5dae918
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_optim.py b/tests/test_optim.py
@@ -12,10 +12,9 @@
 from torch.testing._internal.common_utils import TestCase
 from torch.nn import Parameter
 
-from timm.optim.optim_factory import param_groups_layer_decay, param_groups_weight_decay
-from timm.scheduler import PlateauLRScheduler
-
 from timm.optim import create_optimizer_v2, list_optimizers, get_optimizer_class
+from timm.optim import param_groups_layer_decay, param_groups_weight_decay
+from timm.scheduler import PlateauLRScheduler
 
 import importlib
 import os

diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py
@@ -18,4 +18,5 @@
 from .sgdp import SGDP
 
 from ._optim_factory import list_optimizers, get_optimizer_class, create_optimizer_v2, \
-    create_optimizer, optimizer_kwargs, OptimInfo, OptimizerRegistry
+    create_optimizer, optimizer_kwargs, OptimInfo, OptimizerRegistry
+from ._param_groups import param_groups_layer_decay, param_groups_weight_decay