diff --git a/avalanche/models/mobilenetv1.py b/avalanche/models/mobilenetv1.py index 532901e16..bfdc595aa 100644 --- a/avalanche/models/mobilenetv1.py +++ b/avalanche/models/mobilenetv1.py @@ -22,7 +22,7 @@ try: from pytorchcv.models.mobilenet import DwsConvBlock except Exception: - from pytorchcv.models.common import DwsConvBlock + from pytorchcv.models.common.conv import DwsConvBlock def remove_sequential(network: nn.Module, all_layers: List[nn.Module]):