From e31e5d2d6430d8cd14664d88f680aa8d62ad4726 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Mon, 11 Nov 2024 08:00:05 +0200 Subject: [PATCH] imports --- tests/test_optim.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_optim.py b/tests/test_optim.py index 1ad8baea..66aaadbf 100644 --- a/tests/test_optim.py +++ b/tests/test_optim.py @@ -11,9 +11,11 @@ from copy import deepcopy 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, param_groups_layer_decay, param_groups_weight_decay +from timm.optim import create_optimizer_v2 import importlib import os