Correcting config to match with model parameters.

config.json

@@ -3,7 +3,7 @@
   "batch_norm": true,
   "batch_size": 16,
   "criterion": "mse",
-  "dropout_rate": 0.3,
+  "dropout": 0.3,
   "img_size": 224,
   "lr": 0.004,
   "max_epochs": 30,