make sure dropout_rates are in valid form

Coderx7 · Coderx7 · commit 0f5ebc0b2806 · 2023-02-19T17:56:56.000+03:30
diff --git a/Cifar/models/simplenet.py b/Cifar/models/simplenet.py
@@ -240,8 +240,8 @@ def __init__(
                 (320, 1, 0.0),
             ],
         }
-
-        self.dropout_rates = drop_rates
+        # make sure values are in proper form!
+        self.dropout_rates = {int(key):float(value) for key,value in drop_rates.items()}
         # 15 is the last layer of the network(including two previous pooling layers)
         # basically specifying the dropout rate for the very last layer to be used after the pooling
         self.last_dropout_rate = self.dropout_rates.get(15, 0.0)
diff --git a/ImageNet/simplenet.py b/ImageNet/simplenet.py
@@ -240,8 +240,8 @@ def __init__(
                 (320, 1, 0.0),
             ],
         }
-
-        self.dropout_rates = drop_rates
+        # make sure values are in proper form!
+        self.dropout_rates = {int(key):float(value) for key,value in drop_rates.items()}
         # 15 is the last layer of the network(including two previous pooling layers)
         # basically specifying the dropout rate for the very last layer to be used after the pooling
         self.last_dropout_rate = self.dropout_rates.get(15, 0.0)
