From 69c30a542f9e91a377c2cf783871593d86831542 Mon Sep 17 00:00:00 2001 From: Leo Paillet Date: Wed, 6 Mar 2024 18:33:40 +0100 Subject: [PATCH] data loading and reconst algo for new use case --- data_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_handler.py b/data_handler.py index 3572d5b..53c68e5 100644 --- 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) \ No newline at end of file