fastmri_brain_mini/VORTEX_Image+Motion__super-heavy/config.yaml

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