make it to template

configs/logger/comet.yaml

@@ -4,7 +4,7 @@ comet:
   _target_: lightning.pytorch.loggers.comet.CometLogger
   api_key: ${oc.env:COMET_API_TOKEN} # api key is loaded from environment variable
   save_dir: "${paths.output_dir}"
-  project_name: "fastmri_reconstruction_unet"
+  project_name: "MRI_Recon_fastMRI_Template"
   rest_api_key: null
   # experiment_name: ""
   experiment_key: null # set to resume experiment

configs/train.yaml

@@ -29,7 +29,7 @@ defaults:
   - debug: null
 
 # task name, determines output directory path
-task_name: "fastmri_knee_training"
+task_name: "MRI_Recon_fastMRI_Template"
 
 # tags to help you identify your experiments
 # you can overwrite this in experiment configs

setup.py

@@ -5,10 +5,10 @@
 setup(
     name="src",
     version="0.0.1",
-    description="Describe Your Cool Project",
-    author="",
-    author_email="",
-    url="https://github.com/user/project",
+    description="MRI Reconstruction using Lighting and Hydra",
+    author="Yuyang Xue",
+    author_email="yuyang.xue@ed.ac.uk",
+    url="https://github.com/YuyangXueEd/ReconHydra",
     install_requires=["lightning", "hydra-core"],
     packages=find_packages(),
     # use this to customize global commands available in the terminal after installing the package

src/data/fastmri_datamodule.py

@@ -114,6 +114,7 @@ def __init__(
         if sample_rate is not None and volume_sample_rate is not None:
             raise ValueError("Either sample_rate or volume_sample_rate should be set but not both.")
 
+        Path("../data/caches/").mkdir(parents=True, exist_ok=True)
         self.dataset_cache_file = Path('../data/caches/' + dataset_cache_file)
         self.transform = transform
         self.recons_key = "reconstruction_rss" if challenge == "multicoil" else "reconstruction_esc"
@@ -471,7 +472,7 @@ def __init__(
 
         # this line allows to access init params with 'self.hparams' attribute
         # also ensures init params will be stored in ckpt
-        self.save_hyperparameters(logger=False)
+        self.save_hyperparameters(ignore=['net'])
 
         self.data_path = Path(data_path)
         self.challenge = challenge

src/models/mri_module.py

@@ -30,7 +30,7 @@ def __init__(
 
         # this line allows to access init params with 'self.hparams' attribute
         # also ensures init params will be stored in ckpt
-        self.save_hyperparameters(logger=False)
+        self.save_hyperparameters(ignore=['net'])
 
         self.num_log_images = num_log_images