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

Add guard around 'somewhat' newer torch RAdam / NAdam imports

Browse Source
This commit is contained in:
Ross Wightman 2024-11-26 11:10:42 -08:00 committed by Ross Wightman
parent 0903d98162
commit 4f64ec4e14
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
timm/optim/__init__.py
View File

@ -21,8 +21,13 @@ from .rmsprop_tf import RMSpropTF
from .sgdp import SGDP from .sgdp import SGDP
from .sgdw import SGDW from .sgdw import SGDW
# bring torch optim into timm.optim namespace for consistency # bring common torch.optim Optimizers into timm.optim namespace for consistency
from torch.optim import Adadelta, Adagrad, Adamax, Adam, NAdam, RAdam, RMSprop, SGD from torch.optim import Adadelta, Adagrad, Adamax, Adam, AdamW, RMSprop, SGD
try:
# in case any very old torch versions being used
from torch.optim import NAdam, RAdam
except ImportError:
pass
from ._optim_factory import list_optimizers, get_optimizer_class, get_optimizer_info, OptimInfo, OptimizerRegistry, \ from ._optim_factory import list_optimizers, get_optimizer_class, get_optimizer_info, OptimInfo, OptimizerRegistry, \
create_optimizer_v2, create_optimizer, optimizer_kwargs create_optimizer_v2, create_optimizer, optimizer_kwargs