add 600MHz training

Readme.md

@@ -11,7 +11,7 @@ Python 3.9+
 
 GPU version (for training and inference)
 ```
-pip install -r requirements.txt
+pip install -r requirements-gpu.txt
 ```
 
 CPU version (for inference, not recommended for training)
@@ -159,18 +159,30 @@ dic, data = ng.varian.read(varian_fid_path)
 
 If you want to train the network using the calibration data from our paper, follow the procedure below.
 
-1. Download multiplets database and our SCRF file:
+1. Download multiplets database and our SCRF files:
     ```
     python download_files.py --multiplets --SCRF --no-weights
     ```
+    or directly download from Google Drive and store in `data/` directory: [Response Functions 600MHz](https://drive.google.com/file/d/1J-DsPtaITXU3TFrbxaZPH800U1uIiwje/view?usp=sharing), [Response Functions 700MHz](https://drive.google.com/file/d/113al7A__yYALx_2hkESuzFIDU3feVtNY/view?usp=sharing), [Multiplets data](https://drive.google.com/file/d/1QGvV-Au50ZxaP1vFsmR_auI299Dw-Wrt/view?usp=sharing)
+
 2. Configure run
-    ```bash
-    mkdir -p runs/repeat_paper_training
-    cp configs/shimnet_700.yaml runs/repeat_paper_training/config.yaml
-    ```
+    - For 600MHz spectrometer:
+      ```bash
+      mkdir -p runs/repeat_paper_training_600MHz
+      cp configs/shimnet_600.yaml runs/repeat_paper_training_600MHz/config.yaml
+      ```
+    - For 700 MHz spectrometer:
+      ```bash
+      mkdir -p runs/repeat_paper_training_700MHz
+      cp configs/shimnet_700.yaml runs/repeat_paper_training_700MHz/config.yaml
+      ```
 3. Run training:
     ```
-    python train.py runs/repeat_paper_training
+    python train.py runs/repeat_paper_training_600MHz
+    ```
+    or
+    ```
+    python train.py runs/repeat_paper_training_700MHz
     ```
-    Training results will appear in `runs/repeat_paper_training` directory.
+    Training results will appear in `runs/repeat_paper_training_600MHz` or `runs/repeat_paper_training_700MHz` directory.
 

configs/shimnet_600.yaml

@@ -23,7 +23,7 @@ data:
   # Paste path to your SCRF file here
   # - Can be absolute path
   # - Can be relative to repository root
-  - data/600hz-2-Lnative/scrf_81.pt
+  - data/scrf_81_600MHz.pt
   atom_groups_data_file: data/multiplets_10000_parsed.txt
   response_function_stretch_min: 1.0
   response_function_stretch_max: 1.0

download_files.py

@@ -14,6 +14,10 @@ ALL_FILES_TO_DOWNLOAD = {
     "SCRF": [{
         "url": "https://drive.google.com/uc?export=download&id=113al7A__yYALx_2hkESuzFIDU3feVtNY",
         "destination": "data/scrf_61_700MHz.pt"
+    },
+    {
+        "url": "https://drive.google.com/uc?export=download&id=1J-DsPtaITXU3TFrbxaZPH800U1uIiwje",
+        "destination": "data/scrf_81_600MHz.pt"
     }],
     "mupltiplets": [{
         "url": "https://drive.google.com/uc?export=download&id=1QGvV-Au50ZxaP1vFsmR_auI299Dw-Wrt",
@@ -38,7 +42,7 @@ def parse_args():
         dest='weights',
         help='Do not download weights file.',
     )
-    parser.add_argument('--SCRF', action='store_true', help='Download SCRF file') 
+    parser.add_argument('--SCRF', action='store_true', help='Download SCRF files - Shim Coil Response Functions') 
     parser.add_argument('--multiplets', action='store_true', help='Download multiplets data file')
     parser.add_argument('--development', action='store_true', help='Download development weights file')