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

Remove rogue import, thanks IDE :/

Browse Source
This commit is contained in:
Ross Wightman 2024-11-26 12:20:20 -08:00 committed by Ross Wightman
parent 73d10ab482
commit 7a165fcb62
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_optim.py
View File

@ -3,23 +3,20 @@
These tests were adapted from PyTorch' optimizer tests. These tests were adapted from PyTorch' optimizer tests.
""" """
import math
import pytest
import functools import functools
import importlib
import os
from copy import deepcopy from copy import deepcopy
import pytest
import torch import torch
from IPython.testing.decorators import skip_win32
from torch.testing._internal.common_utils import TestCase
from torch.nn import Parameter from torch.nn import Parameter
from torch.testing._internal.common_utils import TestCase
from timm.optim import create_optimizer_v2, list_optimizers, get_optimizer_class, get_optimizer_info, OptimInfo from timm.optim import create_optimizer_v2, list_optimizers, get_optimizer_class, get_optimizer_info, OptimInfo
from timm.optim import param_groups_layer_decay, param_groups_weight_decay from timm.optim import param_groups_layer_decay, param_groups_weight_decay
from timm.scheduler import PlateauLRScheduler from timm.scheduler import PlateauLRScheduler
import importlib
import os
torch_backend = os.environ.get('TORCH_BACKEND') torch_backend = os.environ.get('TORCH_BACKEND')
if torch_backend is not None: if torch_backend is not None:
importlib.import_module(torch_backend) importlib.import_module(torch_backend)