diff --git a/avalanche/models/mobilenetv1.py b/avalanche/models/mobilenetv1.py index a2bbedb40..486c4d408 100644 --- a/avalanche/models/mobilenetv1.py +++ b/avalanche/models/mobilenetv1.py @@ -25,7 +25,8 @@ try: from pytorchcv.models.common import DwsConvBlock except ImportError: - from pytorchcv.models.common.conv import DwsConvBlock # pytorchcv >= 0.0.68 + # pytorchcv >= 0.0.68 + from pytorchcv.models.common.conv import DwsConvBlock def remove_sequential(network: nn.Module, all_layers: List[nn.Module]):