Fix break of 2 years old torchvision installs :/

rwightman · rwightman · commit 9fab8d8f58e1 · 2023-11-04T02:32:09.000-07:00
diff --git a/timm/layers/create_norm.py b/timm/layers/create_norm.py
@@ -11,7 +11,7 @@
 import torch.nn as nn
 
 from .norm import GroupNorm, GroupNorm1, LayerNorm, LayerNorm2d, RmsNorm
-from torchvision.ops import FrozenBatchNorm2d
+from torchvision.ops.misc import FrozenBatchNorm2d
 
 _NORM_MAP = dict(
     batchnorm=nn.BatchNorm2d,
