Update config.json

config.json

@@ -3,18 +3,47 @@
     "EfficientNetB0"
   ],
   "model_type": "EfficientNetB0",
-  "num_labels": 7,
-  "hidden_size": 128,
+"num_labels": 7,
+  "hidden_size": 300,
   "num_attention_heads": 1,
   "num_hidden_layers": 2,
-  "intermediate_size": 0,
+  "intermediate_size": 1200,
   "hidden_act": "relu",
   "hidden_dropout_prob": 0.2,
   "attention_probs_dropout_prob": 0.2,
   "initializer_range": 0.02,
   "layer_norm_eps": 1e-12,
-  "image size": 128,
+  "image_size": 128,
   "num_channels": 3,
+  "rescale": 0.00392156863,
+  "shear_range": 0.2,
+  "zoom_range": 0.2,
+  "horizontal_flip": true,
+  "batch_size": 64,
+  "learning_rate": 0.001,
+  "loss": "categorical_crossentropy",
+  "metrics": ["accuracy"],
+  "callbacks": [
+    {
+      "type": "EarlyStopping",
+      "params": {
+        "monitor": "val_loss",
+        "mode": "min",
+        "patience": 5,
+        "verbose": 1
+      }
+    },
+    {
+      "type": "ModelCheckpoint",
+      "params": {
+        "filepath": "./best_model_EB0_7_classification.h5",
+        "monitor": "val_loss",
+        "mode": "min",
+        "verbose": 1,
+        "save_best_only": true
+      }
+    }
+  ]
   "label2id": {
     "celeba": 0,
     "celeba_hq": 1,