mridata_knee_3dfse/12x/Supervised+Aug-FT_1sub/config.yaml

AUG_TEST:
  UNDERSAMPLE:
    ACCELERATIONS:
    - 12
AUG_TRAIN:
  MOTION_P: 0.2
  MRI_RECON:
    AUG_SENSITIVITY_MAPS: true
    SCHEDULER_P:
      IGNORE: false
    TRANSFORMS: []
  NOISE_P: 0.2
  UNDERSAMPLE:
    ACCELERATIONS:
    - 12
    CALIBRATION_SIZE: 20
    CENTER_FRACTIONS: []
    MAX_ATTEMPTS: 30
    NAME: PoissonDiskMaskFunc
  USE_MOTION: false
  USE_NOISE: true
CUDNN_BENCHMARK: false
DATALOADER:
  ALT_SAMPLER:
    PERIOD_SUPERVISED: 1
    PERIOD_UNSUPERVISED: 1
  DATA_KEYS: []
  DROP_LAST: true
  FILTER:
    BY: []
  GROUP_SAMPLER:
    AS_BATCH_SAMPLER: false
    BATCH_BY: []
  NUM_WORKERS: 12
  PREFETCH_FACTOR: 2
  SAMPLER_TRAIN: ''
  SUBSAMPLE_TRAIN:
    NUM_TOTAL: 1
    NUM_TOTAL_BY_GROUP: []
    NUM_UNDERSAMPLED: 0
    NUM_VAL: -1
    NUM_VAL_BY_GROUP: []
    SEED: 1000
DATASETS:
  TEST:
  - mridata_knee_2019_test
  TRAIN:
  - mridata_knee_2019_train
  VAL:
  - mridata_knee_2019_val
DESCRIPTION:
  BRIEF: Denoiser finetuning w/ noise aug U-Net 12x - augmentation std_dev=(0.2, 0.5),
    1 supervised scan(s), 0 unsupervised scan(s), 40k iterations, restart-iteration-count
    = True
  ENTITY_NAME: ''
  EXP_NAME: noise2recon/mridata_knee_3dfse/12x/Supervised+Aug-FT_1sub
  PROJECT_NAME: ''
  TAGS:
  - baseline
  - supervised+aug-ft
  - 12x
  - mridata_knee_3dfse
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:
        - 0.2
        - 0.5
    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: /bmrNAS/people/arjun/results/ss_recon/n2r-experiments/denoise-pretrainer/version_010/model_0039199.pth
OUTPUT_DIR: results://noise2recon/mridata_knee_3dfse/12x/Supervised+Aug-FT_1sub
SEED: 1000
SOLVER:
  BASE_LR: 0.001
  BIAS_LR_FACTOR: 1.0
  CHECKPOINT_PERIOD: -20
  GAMMA: 0.1
  GRAD_ACCUM_ITERS: 1
  LR_SCHEDULER_NAME: ''
  MAX_ITER: 40000
  MOMENTUM: 0.9
  OPTIMIZER: Adam
  STEPS:
  - 320000
  TEST_BATCH_SIZE: 16
  TRAIN_BATCH_SIZE: 16
  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: -20
  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: 400