fix 9m config in cfg in simplenet models

Coderx7 · Coderx7 · commit a3616ec12fcf · 2023-02-14T17:05:25.000+03:30
diff --git a/Cifar/models/simplenet.py b/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
 
 
diff --git a/ImageNet/simplenet.py b/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
 
 
diff --git a/ImageNet/training_scripts/imagenet_training/timm/models/simplenet.py b/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"
     ),
 }
