fix plot in training accuracy surrogate

Author: Demoren1

code/dependencies/GCN.py

@@ -512,7 +512,7 @@ def train_model_accuracy(
             f"Valid Loss: {avg_valid_loss * 1e4:.4f}, LR: {lr:.6f}"
         )
 
-    tmp_train_losses = np.array(train_losses)
+    tmp_train_losses = np.array(train_losses) * 1e4
     tmp_valid_losses = np.array(valid_losses) * 1e4
     plot_train_valid_losses(tmp_train_losses, tmp_valid_losses, file_name="accuracy_model.png")
 

code/surrogate_hp.json

@@ -3,9 +3,9 @@
     "num_workers": 4,
     "dataset_path": "datasets/third_dataset",
     "device": "cuda",
-    "developer_mode": true,
+    "developer_mode": false,
 
-    "n_models": 300,
+    "n_models": 1300,
 
     "upper_margin": 0.75,
     "lower_margin": 0.25,
@@ -17,13 +17,13 @@
 
     "acc_num_epochs": 10,
     "acc_lr_start": 1e-2,
-    "acc_lr_end": 5e-4,
+    "acc_lr_end": 1e-3,
     "acc_dropout": 0.2,
     "acc_n_heads": 16,
 
-    "div_num_epochs": 5,
+    "div_num_epochs": 10,
     "div_lr_start": 1e-3,
-    "div_lr_end": 1e-6,
+    "div_lr_end": 1e-4,
     "div_dropout": 0.1,
     "div_n_heads": 4,
     "margin": 1,