Fix break of 2 years old torchvision installs :/

2023-11-04 02:32:09 -07:00 · 2023-11-04 02:32:09 -07:00 · 9fab8d8f58
parent d3e83a190f
commit 9fab8d8f58
1 changed files with 1 additions and 1 deletions
--- a/timm/layers/create_norm.py
+++ b/timm/layers/create_norm.py
@ -11,7 +11,7 @@ from typing import Type
 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,