fix tmm version

Cathy0908 · Oct 27, 2023 · ddbaa4a · ddbaa4a
1 parent e04ce3c
commit ddbaa4a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/easycv/models/backbones/efficientformer.py b/easycv/models/backbones/efficientformer.py
@@ -3,14 +3,15 @@
 import os
 from distutils.version import LooseVersion
 
+import timm
 import torch
 import torch.nn as nn
 from timm.models.layers import DropPath, trunc_normal_
 
 from ..modelzoo import efficientformer as model_urls
 from ..registry import BACKBONES
 
-if LooseVersion(torch.__version__) <= LooseVersion('0.8.2'):
+if LooseVersion(timm.__version__) <= LooseVersion('0.8.2'):
     from timm.models.layers.helpers import to_2tuple
 else:
     from timm.layers.helpers import to_2tuple