dimension check for pretrained model weights (#2984)

* dimension check for pretrained model weights

This PR implements a dimension check, before trying to load pre-trained model weights.

Signed-off-by: neuronflow <[email protected]>

* [MONAI] python code formatting

Signed-off-by: monai-bot <[email protected]>
Signed-off-by: neuronflow <[email protected]>

* multi line split error

Signed-off-by: neuronflow <[email protected]>

Co-authored-by: monai-bot <[email protected]>

Author: neuronflow

monai/networks/nets/densenet.py

@@ -306,7 +306,11 @@ def __init__(
             **kwargs,
         )
         if pretrained:
-            # it only worked when `spatial_dims` is 2
+            if kwargs["spatial_dims"] > 2:
+                raise NotImplementedError(
+                    "Parameter `spatial_dims` is > 2 ; currently PyTorch Hub does not"
+                    "provide pretrained models for more than two spatial dimensions."
+                )
             _load_state_dict(self, "densenet121", progress)
 
 
@@ -329,7 +333,11 @@ def __init__(
             **kwargs,
         )
         if pretrained:
-            # it only worked when `spatial_dims` is 2
+            if kwargs["spatial_dims"] > 2:
+                raise NotImplementedError(
+                    "Parameter `spatial_dims` is > 2 ; currently PyTorch Hub does not"
+                    "provide pretrained models for more than two spatial dimensions."
+                )
             _load_state_dict(self, "densenet169", progress)
 
 
@@ -352,7 +360,11 @@ def __init__(
             **kwargs,
         )
         if pretrained:
-            # it only worked when `spatial_dims` is 2
+            if kwargs["spatial_dims"] > 2:
+                raise NotImplementedError(
+                    "Parameter `spatial_dims` is > 2 ; currently PyTorch Hub does not"
+                    "provide pretrained models for more than two spatial dimensions."
+                )
             _load_state_dict(self, "densenet201", progress)