Post merge fix reference of old param groups helper fn locations

tests/test_optim.py

@@ -12,10 +12,9 @@ import torch
 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

timm/optim/__init__.py

@@ -18,4 +18,5 @@ from .rmsprop_tf import RMSpropTF
 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