model: graph_conv-mlp_h128_l3_edge_prediction | (graph_conv-mlp_h128_l3) | WandB: aqyutqg5

config.json

@@ -65,13 +65,13 @@
     "optimizer": "adamw",
     "scheduler": "onecycle",
     "gradient_clip_val": null,
-    "epochs": 500,
+    "epochs": 1000,
     "batches_per_epoch": 20,
     "accelerator": "auto",
     "devices": 1,
     "precision": 32,
     "early_stopping": true,
-    "early_stopping_patience": 50,
+    "early_stopping_patience": 150,
     "early_stopping_metric": "val_relation_weighted_auc",
     "save_checkpoints": true,
     "checkpoint_subdir": "checkpoints",
@@ -93,7 +93,7 @@
       "hidden_128",
       "layers_3",
       "lr_0.0005",
-      "epochs_500"
+      "epochs_1000"
     ],
     "log_model": false,
     "mode": "online",