diff --git a/best_configs/hail_expB.json b/best_configs/hail_expB.json
index 0fae03f..a93d5e1 100644
--- a/best_configs/hail_expB.json
+++ b/best_configs/hail_expB.json
@@ -11,7 +11,7 @@
         "clip_ratio": 0.2,
         "iterations_per_epoch": 100,
         "hidden_layer_size": 20,
-        "n_hidden_layers": 4,
+        "n_hidden_layers": 2,
         "activation": 0,
         "option": 0
     },
@@ -23,11 +23,11 @@
         "learning_rate": 0.001,
         "weight_decay": 0.0001,
         "iterations_per_epoch": 100,
-        "n_hidden_layers_element": 3,
+        "n_hidden_layers_element": 4,
         "n_hidden_layers_global": 2,
         "hidden_layer_size": 10,
         "activation": 0
     },
-    "train_epochs": 100,
+    "train_epochs": 85,
     "seed": 0
 }
\ No newline at end of file
diff --git a/best_configs/shail_expA.json b/best_configs/shail_expA.json
index 6de9706..76472e0 100644
--- a/best_configs/shail_expA.json
+++ b/best_configs/shail_expA.json
@@ -10,8 +10,8 @@
         "learning_rate_decay": 1.0,
         "clip_ratio": 0.2,
         "iterations_per_epoch": 100,
-        "hidden_layer_size": 20,
-        "n_hidden_layers": 4,
+        "hidden_layer_size": 40,
+        "n_hidden_layers": 2,
         "activation": 0,
         "option": 0
     },
@@ -23,11 +23,11 @@
         "learning_rate": 0.001,
         "weight_decay": 0.0001,
         "iterations_per_epoch": 100,
-        "n_hidden_layers_element": 3,
-        "n_hidden_layers_global": 2,
+        "n_hidden_layers_element": 4,
+        "n_hidden_layers_global": 1,
         "hidden_layer_size": 10,
         "activation": 0
     },
-    "train_epochs": 100,
+    "train_epochs": 90,
     "seed": 0
 }
\ No newline at end of file
diff --git a/best_configs/shail_expB.json b/best_configs/shail_expB.json
index 04139c2..3a35200 100644
--- a/best_configs/shail_expB.json
+++ b/best_configs/shail_expB.json
@@ -11,7 +11,7 @@
         "clip_ratio": 0.2,
         "iterations_per_epoch": 100,
         "hidden_layer_size": 20,
-        "n_hidden_layers": 4,
+        "n_hidden_layers": 2,
         "activation": 0,
         "option": 0
     },
@@ -23,11 +23,11 @@
         "learning_rate": 0.001,
         "weight_decay": 0.0001,
         "iterations_per_epoch": 100,
-        "n_hidden_layers_element": 3,
+        "n_hidden_layers_element": 4,
         "n_hidden_layers_global": 2,
         "hidden_layer_size": 10,
         "activation": 0
     },
-    "train_epochs": 100,
+    "train_epochs": 85,
     "seed": 0
 }
\ No newline at end of file