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

Fix break of 2 years old torchvision installs :/

Browse Source
This commit is contained in:
Ross Wightman 2023-11-04 02:32:09 -07:00
parent d3e83a190f
commit 9fab8d8f58
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/layers/create_norm.py
View File

@ -11,7 +11,7 @@ from typing import Type
import torch.nn as nn import torch.nn as nn
from .norm import GroupNorm, GroupNorm1, LayerNorm, LayerNorm2d, RmsNorm from .norm import GroupNorm, GroupNorm1, LayerNorm, LayerNorm2d, RmsNorm
from torchvision.ops import FrozenBatchNorm2d from torchvision.ops.misc import FrozenBatchNorm2d
_NORM_MAP = dict( _NORM_MAP = dict(
batchnorm=nn.BatchNorm2d, batchnorm=nn.BatchNorm2d,