Fixes to normalization and time args

gnn-tracking · Mar 8, 2024 · 35e15bc · 35e15bc
1 parent 4736517
commit 35e15bc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/scripts/full_detector/configs/gc/model.yml b/scripts/full_detector/configs/gc/model.yml
@@ -17,7 +17,7 @@ init_args:
       p_attr: 2
       p_rep: 2
       r_emb: 1
-      normalization: n_rep_edges
+      rep_normalization: n_rep_edges
   gc_scanner:
     class_path: gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner
     init_args:

diff --git a/scripts/full_detector/configs/oc/model_gc_loss.yml b/scripts/full_detector/configs/oc/model_gc_loss.yml
@@ -30,7 +30,7 @@ init_args:
       p_attr: 2
       p_rep: 2
       r_emb: 1
-      normalization: n_rep_edges
+      rep_normalization: n_rep_edges
   cluster_scanner:
     class_path: gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner
     init_args:

diff --git a/scripts/full_detector/run_oc.py b/scripts/full_detector/run_oc.py
@@ -64,7 +64,7 @@ def cli_main():
             ],
             "logger": [tb_logger, logger],
             "plugins": [SLURMEnvironment(auto_requeue=False)],
-            "max_time": "23:30:00",
+            "max_time": "0:23:30:00",
         },
     )