Multi-Input-Output

Author: gregbruss

Diff for: .gitignore

@@ -14,3 +14,4 @@ add_header.py
 *fair_cluster*
 fastmri_dirs.yaml
 dataset_cache.pkl
+devnotes.md

Diff for: experimental/varnet/train_varnet_demo.py

@@ -48,6 +48,7 @@ def build_args():
     # ------------------------
     path_config = pathlib.Path.cwd() / ".." / ".." / "fastmri_dirs.yaml"
     knee_path = fetch_dir("knee_path", path_config)
+    brain_path = fetch_dir("brain_path", path_config)
     logdir = fetch_dir("log_path", path_config) / "varnet" / "varnet_demo"
 
     parent_parser = ArgumentParser(add_help=False)
@@ -73,7 +74,7 @@ def build_args():
         lr_step_size=40,
         lr_gamma=0.1,
         weight_decay=0.0,
-        data_path=knee_path,
+        data_path=brain_path,
         challenge="multicoil",
         exp_dir=logdir,
         exp_name="varnet_demo",
@@ -90,6 +91,7 @@ def build_args():
         distributed_backend=backend,
         seed=42,
         deterministic=True,
+        overfit_batches=1
     )
 
     parser.add_argument("--mode", default="train", type=str)

Diff for: fastmri/mri_module.py

@@ -183,9 +183,27 @@ def validation_step_end(self, val_logs):
         val_logs = {key: value.cpu() for key, value in val_logs.items()}
         val_logs["device"] = device
 
+        print("End of Validation Step has been reached!...Saving this val log now!")
+        
+        for k, v in val_logs.items():
+            if k == "device":
+                continue
+            print(k, v.ndim)
+
         return val_logs
 
     def validation_epoch_end(self, val_logs):
+        
+        for k, v in val_logs[0].items():
+            if k == "device":
+                continue
+            print(k, val_logs[0][f"{k}"].ndim)
+
+        
+        for output in val_logs[0]["output"]:
+            print("outputs: ", len(output))
+
+
         assert val_logs[0]["output"].ndim == 3
         device = val_logs[0]["device"]