WIP

gnn-tracking · Nov 6, 2023 · 3d1ed51 · 3d1ed51
1 parent d53255e
commit 3d1ed51
Show file tree

Hide file tree

Showing 2 changed files with 473 additions and 180 deletions.
diff --git a/notebooks/030_performance.ipynb b/notebooks/030_performance.ipynb
diff --git a/src/ocpaper231/data.py b/src/ocpaper231/data.py
@@ -8,15 +8,16 @@
 def get_dm(*, n_val=5, setup=True) -> TrackingDataModule:
     """Get default tracking data module"""
     dm = TrackingDataModule(
+        identifier="point_clouds_v8",
         train={
             "dirs": [
-                "/scratch/gpfs/IOJALVO/gnn-tracking/object_condensation/point_clouds_v6/part_1/"
+                "/scratch/gpfs/IOJALVO/gnn-tracking/object_condensation/point_clouds_v8/part_1/"
             ],
             # If you run into memory issues, reduce this
         },
         val={
             "dirs": [
-                "/scratch/gpfs/IOJALVO/gnn-tracking/object_condensation/point_clouds_v6/part_9/"
+                "/scratch/gpfs/IOJALVO/gnn-tracking/object_condensation/point_clouds_v8/part_9/"
             ],
             "stop": n_val,
         },