Updated import for pytorchcv>=0.0.68

ContinualAI · Oct 28, 2024 · 972162c · 972162c
1 parent d752103
commit 972162c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/avalanche/models/mobilenetv1.py b/avalanche/models/mobilenetv1.py
@@ -21,8 +21,11 @@
 
 try:
     from pytorchcv.models.mobilenet import DwsConvBlock
-except Exception:
-    from pytorchcv.models.common import DwsConvBlock
+except ImportError:
+    try:
+        from pytorchcv.models.common import DwsConvBlock 
+    except ImportError:
+        from pytorchcv.models.common.conv import DwsConvBlock  # pytorchcv >= 0.0.68
 
 
 def remove_sequential(network: nn.Module, all_layers: List[nn.Module]):