mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

imports

pull/2333/head
mrT23 2024-11-11 08:00:05 +02:00 committed by Ross Wightman
parent 68d5a64e45
commit e31e5d2d64
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_optim.py
View File

@ -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