fix unknow keyword error caused by drop_rate keyword being sent by timm

Coderx7 · Coderx7 · commit 22142e6718b0 · 2023-02-14T16:35:03.000+03:30
diff --git a/ImageNet/training_scripts/imagenet_training/timm/models/simplenet.py b/ImageNet/training_scripts/imagenet_training/timm/models/simplenet.py
@@ -436,7 +436,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
 
 
