alpha release of vortex models

.gitattributes

@@ -25,3 +25,95 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zstandard filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion__super-heavy/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion__super-heavy/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion__super-heavy/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Aug_Motion__super-heavy/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/MRAugment_Image/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/MRAugment_Image/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/MRAugment_Image/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/MRAugment_Image/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Noise2Recon_super-heavy/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Noise2Recon_super-heavy/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Noise2Recon_super-heavy/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Noise2Recon_super-heavy/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/SSDU/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/SSDU/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/SSDU/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/SSDU/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Supervised/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Supervised/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Supervised/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/Supervised/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion/model.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion__super-heavy/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion__super-heavy/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion__super-heavy/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+fastmri_brain_mini/VORTEX_Motion__super-heavy/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Image+Physics/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Image+Physics/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Image+Physics/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Image+Physics/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Motion/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Motion/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Motion/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Motion/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Noise/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Noise/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Noise/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Noise/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Physics/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Physics/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Physics/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Aug_Physics/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/MRAugment_Image/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/MRAugment_Image/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/MRAugment_Image/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/MRAugment_Image/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/SSDU/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/SSDU/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/SSDU/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/SSDU/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Supervised/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Supervised/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Supervised/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/Supervised/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Motion/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Motion/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Motion/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Motion/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Noise/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Noise/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Noise/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Noise/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Physics/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Physics/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Physics/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Curricula_Physics/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image+Physics/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image+Physics/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image+Physics/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image+Physics/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Image/model.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Physics/model-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Physics/model-raw-gpu.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Physics/model-raw.ckpt filter=lfs diff=lfs merge=lfs -text
+mridata_knee_3dfse/VORTEX_Physics/model.ckpt filter=lfs diff=lfs merge=lfs -text

fastmri_brain_mini/Aug_Motion/config.yaml

@@ -0,0 +1,227 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+      gamma: 5.0
+      max_iter: -200
+      name: WarmupStepTF
+      warmup_method: exp
+      warmup_milestones:
+      - -1
+    TRANSFORMS:
+    - name: RandomMRIMotion
+      p: 0.2
+      std_devs:
+      - 0.2
+      - 0.5
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: GroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 18
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 0
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(MRAugment\) - Exp:
+    fastmri brain - motion/heavy/version_009 -  MRAugment + motion, heavy, p=0.2'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/Aug_Motion
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - aug
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS: []
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: UnetModel
+  N2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/Aug_Motion
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/Aug_Motion/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c2bded991652d43b63878ae5637c43909450e3d4c2d30d4dfb8b478ec153a65
+size 93109239

fastmri_brain_mini/Aug_Motion/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ef44b5a909c62f2a4fbf020cdc668539fc079750a2930a7116d6126a84da778
+size 93109239

fastmri_brain_mini/Aug_Motion/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffd4b649b0a84f236ca2f86e7588cacbd30b496a961017bef9b1db9903812c31
+size 93103415

fastmri_brain_mini/Aug_Motion/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffd4b649b0a84f236ca2f86e7588cacbd30b496a961017bef9b1db9903812c31
+size 93103415

fastmri_brain_mini/Aug_Motion__super-heavy/config.yaml

@@ -0,0 +1,228 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+      gamma: 5.0
+      max_iter: -200
+      name: WarmupStepTF
+      warmup_method: exp
+      warmup_milestones:
+      - -1
+    TRANSFORMS:
+    - name: RandomMRIMotion
+      p: 0.2
+      std_devs:
+      - 0.5
+      - 0.7
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: GroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 18
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 0
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(MRAugment\) - Exp:
+    fastmri brain - motion/super-heavy/version_009 -  MRAugment + motion, super-heavy,
+    p=0.2'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/Aug_Motion__super-heavy
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - aug
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS: []
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: UnetModel
+  N2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/Aug_Motion__super-heavy
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/Aug_Motion__super-heavy/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7daf8e22659bfa26015b7668691331f6e2bb667598e2e8fe34fda5caf481bd97
+size 93109239

fastmri_brain_mini/Aug_Motion__super-heavy/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32b010785e07dcd3e87d354036525d6b7d8414d60396e6bb1daf8045418ca78c
+size 93109239

fastmri_brain_mini/Aug_Motion__super-heavy/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a31448787ff97797c01a46626f30c40e2a70920494aea64a51f3f6ee27801a54
+size 93103415

fastmri_brain_mini/Aug_Motion__super-heavy/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a31448787ff97797c01a46626f30c40e2a70920494aea64a51f3f6ee27801a54
+size 93103415

fastmri_brain_mini/MRAugment_Image/config.yaml

@@ -0,0 +1,247 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+      gamma: 5.0
+      max_iter: -200
+      name: WarmupStepTF
+      warmup_method: exp
+      warmup_milestones:
+      - -1
+    TRANSFORMS:
+    - dims: -1
+      name: RandomFlip
+      p: 0.275
+    - dims: -2
+      name: RandomFlip
+      p: 0.275
+    - ks:
+      - 2
+      name: RandomRot90
+      p: 0.275
+    - name: RandomTranslation
+      p: 0.55
+      pad_mode: reflect
+      translate:
+      - 0.08
+      - 0.125
+    - angle: 180.0
+      name: RandomAffine
+      p:
+        angle: 0.275
+        scale: 0.55
+        shear: 0.55
+      pad_like: MRAugment
+      scale: 0.25
+      shear: 15.0
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: GroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 18
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 0
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(MRAugment\) - Exp:
+    fastmri brain - image/version_008 -  MRAugment + image, light, p=0.2'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/MRAugment_Image
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - mraugment
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS: []
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: UnetModel
+  N2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/MRAugment_Image
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/MRAugment_Image/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:940b8db5d9742bdb070d9bdaf3c78afab741b5781678bdb5aa8bee15a3053949
+size 93109239

fastmri_brain_mini/MRAugment_Image/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6462f8465ce571cbc0770b8f50a8236bac0e48079500204bd7fb5803a1bab119
+size 93109239

fastmri_brain_mini/MRAugment_Image/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21c561c74189d3cfdeb2af76084e69d29be35226913444b8d4b44cca6842d08d
+size 93103415

fastmri_brain_mini/MRAugment_Image/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21c561c74189d3cfdeb2af76084e69d29be35226913444b8d4b44cca6842d08d
+size 93103415

fastmri_brain_mini/Noise2Recon_super-heavy/config.yaml

@@ -0,0 +1,222 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: AlternatingGroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 36
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini (Aug2Recon) - noise/super-heavy/version_001
+    -  Data ratio - 18 supervised : 36 unsupervised, 1:1 alt sampling, noise/super-heavy/version_001'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/Noise2Recon_super-heavy
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - noise2recon
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS:
+        - name: RandomNoise
+          p: 1.0
+          std_devs:
+          - 0.5
+          - 0.7
+          use_mask: true
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: A2RModel
+  N2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/Noise2Recon_super-heavy
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: 0
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/Noise2Recon_super-heavy/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77762f28bc196b6c747210ddc6016232d50d7bb016107e6d0d5541d921e8ac9a
+size 93110071

fastmri_brain_mini/Noise2Recon_super-heavy/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:19157a4f1e79aa91da5d7ce65268bc546c29e4f1ec6ab594ea88b9eed98e7382
+size 93110071

fastmri_brain_mini/Noise2Recon_super-heavy/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbef095285933a9499dac8d6c1ff4e90382df53108021dba97cf739783658d42
+size 93103607

fastmri_brain_mini/Noise2Recon_super-heavy/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbef095285933a9499dac8d6c1ff4e90382df53108021dba97cf739783658d42
+size 93103607

fastmri_brain_mini/SSDU/config.yaml

@@ -0,0 +1,220 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: GroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 18
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: SSDU U-Net 8x fastMRI brain-mc mini - rho=0.2, uniform/rho=2e-01/version_009
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/SSDU
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - ssdu
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS: []
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: SSDUModel
+  N2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: k_l1_l2_sum_normalized
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS:
+        kind: uniform
+        p: 1.0
+        per_example: true
+        rhos: 0.2
+        std_scale: 4
+    META_ARCHITECTURE: UnetModel
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/SSDU
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 3
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 8
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/SSDU/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d55d05b9be7bd5d1c02b0059dff576f86e331e47bf668f8058093bfe3816a4e
+size 93110071

fastmri_brain_mini/SSDU/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c71257f00538451de2264f62e31aeb2cee35a46f2ea75d95713624fb7a9e7c7c
+size 93110071

fastmri_brain_mini/SSDU/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:588b65426755ec783907d2435c3b26fa0789e54ef0b2b7a37761b67fbc68e10a
+size 93103607

fastmri_brain_mini/SSDU/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:588b65426755ec783907d2435c3b26fa0789e54ef0b2b7a37761b67fbc68e10a
+size 93103607

fastmri_brain_mini/Supervised/config.yaml

@@ -0,0 +1,215 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: GroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 18
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 0
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini baseline
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/Supervised
+  PROJECT_NAME: ''
+  TAGS:
+  - baseline
+  - supervised
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS: []
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: UnetModel
+  N2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/Supervised
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 400

fastmri_brain_mini/Supervised/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa8537512d781c843465cac20d51deede67fbb16f17aa20a3d1155f7ed3ae577
+size 93109239

fastmri_brain_mini/Supervised/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fc1022d8b6744d8d701744c9b0993cd707088a55874b86d3115cbc45c914bcc
+size 93109239

fastmri_brain_mini/Supervised/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d10a8fb756af50946d712b1e5174e88f5e2c8fd6ed21abbc70bbb6913d859f7d
+size 93103415

fastmri_brain_mini/Supervised/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d10a8fb756af50946d712b1e5174e88f5e2c8fd6ed21abbc70bbb6913d859f7d
+size 93103415

fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/config.yaml

@@ -0,0 +1,246 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: AlternatingGroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 36
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(Aug2Recon\) - 18sup36unsup/super-heavy/version_030
+    -  Data ratio - 18 supervised : 36 unsupervised, 1:1 alt sampling, 18sup36unsup/super-heavy/version_030'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/VORTEX_Image+Motion__super-heavy
+  PROJECT_NAME: ''
+  TAGS:
+  - vortex
+  - vortex
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS:
+        - dims: -1
+          name: RandomFlip
+          p: 0.5
+        - dims: -2
+          name: RandomFlip
+          p: 0.5
+        - ks:
+          - 2
+          name: RandomRot90
+          p: 0.5
+        - name: RandomTranslation
+          p: 1.0
+          pad_mode: reflect
+          translate:
+          - 0.1
+          - 0.1
+        - angle: 180.0
+          name: RandomAffine
+          p:
+            angle: 0.5
+            scale: 1.0
+            shear: 1.0
+          pad_like: MRAugment
+          scale: 0.25
+          shear: 15.0
+        - name: RandomMRIMotion
+          p: 1.0
+          std_devs:
+          - 0.5
+          - 0.7
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: A2RModel
+  N2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/VORTEX_Image+Motion__super-heavy
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2caf870e4f520cc7624becbb0fdd95006316a5bb32d9eb0e9d8ff58ffc6ac424
+size 93110071

fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d995b5c5d923dc17f9914508bc2a46564a4088cf3fe9e6bcf1af10cfa13b8860
+size 93110071

fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bff226d3d6abbf0fb48df2ba673b9d85c7f128ce919fa040328f1567f14729c
+size 93103607

fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bff226d3d6abbf0fb48df2ba673b9d85c7f128ce919fa040328f1567f14729c
+size 93103607

fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/config.yaml

@@ -0,0 +1,252 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: AlternatingGroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 36
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(Aug2Recon\) - 18sup36unsup/super-heavy/version_029
+    -  Data ratio - 18 supervised : 36 unsupervised, 1:1 alt sampling, 18sup36unsup/super-heavy/version_029'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/VORTEX_Image+Physics__super-heavy
+  PROJECT_NAME: ''
+  TAGS:
+  - vortex
+  - vortex
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS:
+        - dims: -1
+          name: RandomFlip
+          p: 0.5
+        - dims: -2
+          name: RandomFlip
+          p: 0.5
+        - ks:
+          - 2
+          name: RandomRot90
+          p: 0.5
+        - name: RandomTranslation
+          p: 1.0
+          pad_mode: reflect
+          translate:
+          - 0.1
+          - 0.1
+        - angle: 180.0
+          name: RandomAffine
+          p:
+            angle: 0.5
+            scale: 1.0
+            shear: 1.0
+          pad_like: MRAugment
+          scale: 0.25
+          shear: 15.0
+        - name: RandomMRIMotion
+          p: 1.0
+          std_devs:
+          - 0.5
+          - 0.7
+        - name: RandomNoise
+          p: 1.0
+          std_devs:
+          - 0.5
+          - 0.7
+          use_mask: true
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: A2RModel
+  N2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/VORTEX_Image+Physics__super-heavy
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3da2867cae6908f3538973246146aba5e779d01c580501ec876e499dc06d82d8
+size 93110071

fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:767a0836904121e0cd15fffd314b478eb1a2c5aa215b63eeb33c0f59c13b3b5f
+size 93110071

fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0549c1ac6ee0662280038956156d4356ac5cc749d08ad857eaa4ad030ecf8a20
+size 93103607

fastmri_brain_mini/VORTEX_Image+Physics__super-heavy/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0549c1ac6ee0662280038956156d4356ac5cc749d08ad857eaa4ad030ecf8a20
+size 93103607

fastmri_brain_mini/VORTEX_Motion/config.yaml

@@ -0,0 +1,221 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: AlternatingGroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 18
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(Aug2Recon\) - Spatial
+    Transforms
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/VORTEX_Motion
+  PROJECT_NAME: ''
+  TAGS:
+  - vortex
+  - vortex
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS:
+        - name: RandomMRIMotion
+          p: 1.0
+          std_devs:
+          - 0.2
+          - 0.5
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: A2RModel
+  N2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/VORTEX_Motion
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/VORTEX_Motion/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f70e896c274c0d87d3ccd0395c7d96d5b8db309c564941457a97443ed9294890
+size 93110071

fastmri_brain_mini/VORTEX_Motion/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9d27f35e087b8b1ec2dd4db248315c5812f85537aed878fd42e3858b1274a87
+size 93110071

fastmri_brain_mini/VORTEX_Motion/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ee156063542141ac3032685d431cf63b0697308d31d781b31b1efa1c042dd7f
+size 93103607

fastmri_brain_mini/VORTEX_Motion/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ee156063542141ac3032685d431cf63b0697308d31d781b31b1efa1c042dd7f
+size 93103607

fastmri_brain_mini/VORTEX_Motion__super-heavy/config.yaml

@@ -0,0 +1,221 @@
+AUG_TEST:
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+AUG_TRAIN:
+  MOTION_P: 0.2
+  MRI_RECON:
+    AUG_SENSITIVITY_MAPS: true
+    SCHEDULER_P:
+      IGNORE: false
+    TRANSFORMS: []
+  NOISE_P: 0.2
+  UNDERSAMPLE:
+    ACCELERATIONS:
+    - 8
+    CALIBRATION_SIZE: 0
+    CENTER_FRACTIONS:
+    - 0.04
+    MAX_ATTEMPTS: 30
+    NAME: RandomMaskFunc1D
+  USE_MOTION: false
+  USE_NOISE: false
+CUDNN_BENCHMARK: false
+DATALOADER:
+  ALT_SAMPLER:
+    PERIOD_SUPERVISED: 1
+    PERIOD_UNSUPERVISED: 1
+  DATA_KEYS:
+  - - target
+    - jsense-8-cf=4/target
+  - - maps
+    - jsense-8-cf=4/maps
+  DROP_LAST: true
+  FILTER:
+    BY:
+    - - acquisition
+      - AXT2
+    - - systemFieldStrength_T
+      - 2.8936
+  GROUP_SAMPLER:
+    AS_BATCH_SAMPLER: true
+    BATCH_BY:
+    - receiverChannels
+    - inplane_shape
+  NUM_WORKERS: 8
+  PREFETCH_FACTOR: 2
+  SAMPLER_TRAIN: AlternatingGroupSampler
+  SUBSAMPLE_TRAIN:
+    NUM_TOTAL: 54
+    NUM_TOTAL_BY_GROUP: []
+    NUM_UNDERSAMPLED: 36
+    NUM_VAL: 50
+    NUM_VAL_BY_GROUP: []
+    SEED: 1000
+DATASETS:
+  TEST:
+  - fastMRI_brain_multicoil_mini_v0.0.1_test
+  TRAIN:
+  - fastMRI_brain_multicoil_mini_v0.0.1_train
+  VAL:
+  - fastMRI_brain_multicoil_mini_v0.0.1_val
+DESCRIPTION:
+  BRIEF: 'U-NET LR = 1e-3 Aug2Recon 8x fastMRI brain-mc mini \(Aug2Recon\) - 18sup36unsup/super-heavy/version_032
+    -  Data ratio - 18 supervised : 36 unsupervised, 1:1 alt sampling, 18sup36unsup/super-heavy/version_032'
+  ENTITY_NAME: ''
+  EXP_NAME: vortex/fastmri_brain_mini/VORTEX_Motion__super-heavy
+  PROJECT_NAME: ''
+  TAGS:
+  - vortex
+  - vortex
+  - 8x
+  - fastmri_brain_mini
+MODEL:
+  A2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  CONSISTENCY:
+    AUG:
+      MOTION:
+        RANGE:
+        - 0.2
+        - 0.5
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+      MRI_RECON:
+        AUG_SENSITIVITY_MAPS: true
+        SCHEDULER_P:
+          IGNORE: false
+        TRANSFORMS:
+        - name: RandomMRIMotion
+          p: 1.0
+          std_devs:
+          - 0.5
+          - 0.7
+      NOISE:
+        MASK:
+          RHO: 1.0
+        SCHEDULER:
+          WARMUP_ITERS: 0
+          WARMUP_METHOD: ''
+        STD_DEV:
+        - 1
+    LATENT_LOSS_NAME: mag_l1
+    LATENT_LOSS_WEIGHT: 0.1
+    LOSS_NAME: l1
+    LOSS_WEIGHT: 0.1
+    NUM_LATENT_LAYERS: 1
+    USE_CONSISTENCY: true
+    USE_LATENT: false
+  CS:
+    MAX_ITER: 200
+    REGULARIZATION: 0.005
+  DENOISING:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    NOISE:
+      STD_DEV:
+      - 1
+      USE_FULLY_SAMPLED_TARGET: true
+      USE_FULLY_SAMPLED_TARGET_EVAL: null
+  DEVICE: cpu
+  M2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  META_ARCHITECTURE: A2RModel
+  N2R:
+    META_ARCHITECTURE: UnetModel
+    USE_SUPERVISED_CONSISTENCY: false
+  NM2R:
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+    USE_SUPERVISED_CONSISTENCY: false
+  NORMALIZER:
+    KEYWORDS: []
+    NAME: TopMagnitudeNormalizer
+  RECON_LOSS:
+    NAME: l1
+    RENORMALIZE_DATA: false
+  SEG:
+    ACTIVATION: sigmoid
+    CLASSES: []
+    INCLUDE_BACKGROUND: false
+  SSDU:
+    MASKER:
+      PARAMS: {}
+    META_ARCHITECTURE: GeneralizedUnrolledCNN
+  UNET:
+    BLOCK_ORDER:
+    - conv
+    - relu
+    - conv
+    - relu
+    - batchnorm
+    - dropout
+    CHANNELS: 32
+    DROPOUT: 0.0
+    IN_CHANNELS: 2
+    NORMALIZE: false
+    NUM_POOL_LAYERS: 4
+    OUT_CHANNELS: 2
+  UNROLLED:
+    BLOCK_ARCHITECTURE: ResNet
+    CONV_BLOCK:
+      ACTIVATION: relu
+      NORM: none
+      NORM_AFFINE: false
+      ORDER:
+      - norm
+      - act
+      - drop
+      - conv
+    DROPOUT: 0.0
+    FIX_STEP_SIZE: false
+    KERNEL_SIZE:
+    - 3
+    NUM_EMAPS: 1
+    NUM_FEATURES: 256
+    NUM_RESBLOCKS: 2
+    NUM_UNROLLED_STEPS: 5
+    PADDING: ''
+    SHARE_WEIGHTS: false
+  WEIGHTS: ''
+OUTPUT_DIR: results://vortex/fastmri_brain_mini/VORTEX_Motion__super-heavy
+SEED: 1000
+SOLVER:
+  BASE_LR: 0.001
+  BIAS_LR_FACTOR: 1.0
+  CHECKPOINT_PERIOD: -10
+  GAMMA: 0.1
+  GRAD_ACCUM_ITERS: 2
+  LR_SCHEDULER_NAME: ''
+  MAX_ITER: -200
+  MOMENTUM: 0.9
+  OPTIMIZER: Adam
+  STEPS: []
+  TEST_BATCH_SIZE: 16
+  TRAIN_BATCH_SIZE: 12
+  WARMUP_FACTOR: 0.001
+  WARMUP_ITERS: 1000
+  WARMUP_METHOD: linear
+  WEIGHT_DECAY: 0.0001
+  WEIGHT_DECAY_BIAS: 0.0001
+  WEIGHT_DECAY_NORM: 0.0
+TEST:
+  EVAL_PERIOD: -10
+  EXPECTED_RESULTS: []
+  FLUSH_PERIOD: -10
+  VAL_AS_TEST: true
+  VAL_METRICS:
+    RECON:
+    - psnr
+    - psnr_scan
+    - psnr_mag
+    - psnr_mag_scan
+    - nrmse
+    - nrmse_scan
+    - nrmse_mag
+    - nrmse_mag_scan
+    - ssim (Wang)
+TIME_SCALE: iter
+VERSION: 1
+VIS_PERIOD: 100

fastmri_brain_mini/VORTEX_Motion__super-heavy/model-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5e8586d40e60dd69a5d8bb67b5fbb506d89d6682c6a3840b760140264d40877
+size 93110071

fastmri_brain_mini/VORTEX_Motion__super-heavy/model-raw-gpu.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:95a82cc67e177121023aead06465be4cf9bb95463a51ade8a62cd0514e7b0803
+size 93110071

fastmri_brain_mini/VORTEX_Motion__super-heavy/model-raw.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:727269d95580626fe402eb59054538cddcfd71099bf23674aaefe6adc8f224be
+size 93103607