model: graph_conv-distmult_h128_l3_edge_prediction | (graph_conv-distmult_h128_l3) | WandB: kjv3q37g

config.json

@@ -2,7 +2,7 @@
   "name": null,
   "seed": 42,
   "deterministic": true,
-  "output_dir": "relation_prediction_distmult_128e_transfer_v1",
+  "output_dir": "relation_prediction_distmult_128e_transfer_v2",
   "model": {
     "encoder": "graph_conv",
     "hidden_channels": 128,
@@ -28,9 +28,9 @@
     "edge_encoder_init_bias": 0.0,
     "use_pretrained_model": true,
     "pretrained_model_source": "huggingface",
-    "pretrained_model_path": "seanhacks/edge_prediction_dotprod_128e",
-    "pretrained_model_revision": null,
-    "pretrained_model_load_head": false,
+    "pretrained_model_path": "seanhacks/relation_prediction_distmult_128e",
+    "pretrained_model_revision": "20251228",
+    "pretrained_model_load_head": true,
     "pretrained_model_freeze_encoder_weights": false,
     "pretrained_model_freeze_head_weights": false
   },
@@ -63,7 +63,7 @@
     "lr": 0.0005,
     "weight_decay": 0.0001,
     "optimizer": "adamw",
-    "scheduler": "onecycle",
+    "scheduler": "plateau",
     "gradient_clip_val": null,
     "epochs": 500,
     "batches_per_epoch": 20,