Update example for MAx

CambridgeCIA · Mar 19, 2024 · 4a0f8c2 · 4a0f8c2
1 parent b80d057
commit 4a0f8c2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/LION/experiments/ct_benchmarking_experiments.py b/LION/experiments/ct_benchmarking_experiments.py
@@ -47,4 +47,14 @@ def default_parameters(dataset="2DeteCT"):
 
         # leave this untouched
         param.data_loader_params = Experiment.get_dataset_parameters(dataset)
+
+        # Change the data loader to be recon2recon
+        param.data_loader_params.task = "recon2recon"
+        param.data_loader_params.input_mode = "mode1"
+        param.data_loader_params.target_mode = "mode2"
+
+        # The above only sets the parameters for the data loader, but the data loader is not created yet.
+        # it is created on demand, when the users goes
+        # training_data = ExampleClassForMax.get_training_data()
+
         return param