fix 9m config in cfg in simplenet models

Author: Coderx7

Cifar/models/simplenet.py

@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):
     "simplenetv1_5m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
     ),
-    "simplenetv1_m1_9m": _cfg(
+    "simplenetv1_9m_m1": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
     ),
-    "simplenetv1_m2_9m": _cfg(
+    "simplenetv1_9m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
     ),
 }
@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:
     Returns:
         Dict[str,Any]: cleaned kwargs
     """
-    model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode"]}
+    model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode", "drop_rate"]}
     return model_args
 
 

ImageNet/simplenet.py

@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):
     "simplenetv1_5m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
     ),
-    "simplenetv1_m1_9m": _cfg(
+    "simplenetv1_9m_m1": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
     ),
-    "simplenetv1_m2_9m": _cfg(
+    "simplenetv1_9m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
     ),
 }
@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:
     Returns:
         Dict[str,Any]: cleaned kwargs
     """
-    model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode"]}
+    model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode", "drop_rate"]}
     return model_args
 
 

ImageNet/training_scripts/imagenet_training/timm/models/simplenet.py

@@ -76,10 +76,10 @@ def _cfg(url="", **kwargs):
     "simplenetv1_5m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
     ),
-    "simplenetv1_m1_9m": _cfg(
+    "simplenetv1_9m_m1": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
     ),
-    "simplenetv1_m2_9m": _cfg(
+    "simplenetv1_9m_m2": _cfg(
         url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
     ),
 }