data loading and reconst algo for new use case

a-rouxel · Mar 6, 2024 · 69c30a5 · 69c30a5
1 parent c7392a3
commit 69c30a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_handler.py b/data_handler.py
@@ -201,5 +201,5 @@ def arguement_2(generate_gt):
 #         return acq, cube
 
 if __name__ == "__main__":
-    data_dir = "/local/users/ademaio/lpaillet/mst_datasets"
+    data_dir = "/local/users/ademaio/lpaillet/mst_datasets/cave_1024_28/"
     datamodule = CubesDataModule(data_dir, batch_size=5, num_workers=2)