Test version of run_gc

scripts/pixel/configs/gc/model.yml

@@ -11,7 +11,7 @@ init_args:
   loss_fct:
     class_path: gnn_tracking.metrics.losses.metric_learning.GraphConstructionHingeEmbeddingLoss
     init_args:
-      lw_repulsive: 0.5
+      lw_repulsive: 0.1
       pt_thld: 0.9
       max_num_neighbors: 256
       p_attr: 2

scripts/pixel/run_gc.py

@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 import torch
 from gnn_tracking.training.callbacks import ExpandWandbConfig, PrintValidationMetrics
 from gnn_tracking.utils.loading import TrackingDataModule

scripts/pixel/run_gc_test.py

@@ -0,0 +1,25 @@
+import torch
+from gnn_tracking.training.callbacks import PrintValidationMetrics
+from gnn_tracking.utils.loading import TrackingDataModule
+from pytorch_lightning.callbacks import RichProgressBar
+
+from hpo2.lightning_utils import TorchCompileCLI
+
+
+def cli_main():
+    torch.set_float32_matmul_precision("medium")
+
+    # noinspection PyUnusedLocal
+    cli = TorchCompileCLI(  # noqa: F841
+        datamodule_class=TrackingDataModule,
+        trainer_defaults={
+            "callbacks": [
+                RichProgressBar(leave=True),
+                PrintValidationMetrics(),
+            ],
+        },
+    )
+
+
+if __name__ == "__main__":
+    cli_main()