mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Post merge fix reference of old param groups helper fn locations

Browse Source
This commit is contained in:
Ross Wightman 2024-11-12 16:55:51 -08:00 committed by Ross Wightman
parent 53657a31b7
commit 45490ac52f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_optim.py
View File

@ -12,10 +12,9 @@ import torch
from torch.testing._internal.common_utils import TestCase from torch.testing._internal.common_utils import TestCase
from torch.nn import Parameter 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 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 importlib
import os import os

3
timm/optim/__init__.py
View File

@ -18,4 +18,5 @@ from .rmsprop_tf import RMSpropTF
from .sgdp import SGDP from .sgdp import SGDP
from ._optim_factory import list_optimizers, get_optimizer_class, create_optimizer_v2, \ 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