config_seg_mp_morph.yaml

manual_seed: 2020
GPUs: [0]
base_dir_prev: seg/SegResidualUNet3D
description: morph_mp2_lay4_group4

# model configuration
model:
  name: ResidualMPUNet3D
  in_channels: 1
  out_channels: 18
  f_maps: [16, 32, 64,128]
  layer_order: cgr
  num_groups: 4
  mpb_nums: 2
Morphmodel:
  name: VxmDense
  inshape: [128, 176, 128]
  nb_unet_features: [[16, 32, 32, 32],
                     [32, 32, 32, 32, 32, 16, 16]]
  bidir: false

# trainer configuration
trainer:
  trainer_name: TrainerSegmentMPMorph
  validate_after_iters: 100
  log_after_iters: 10
  stop_after_nobest_iters: 7
  epochs: 200
  eval_score_higher_is_better: true
  amp: false

# loss function configuration
loss:
  name: BCEDiceLoss
  sigmoid_normalization: False

optimizer:
  learning_rate: 0.0001
  weight_decay: 0.0001

# evaluation metric
eval_metric:
  name: DiceCoefficient
Morphloss:
  [name: NCC,name: Grad]

# learning rate scheduler configuration
# lr_scheduler:
#   name: ReduceLROnPlateau
#   mode: min
#   factor: 0.5
#   patience: 5

lr_scheduler:
  name: StepLR
  step_size: 5
  gamma: 0.8

# data loaders configuration
loaders:
  csv_path: DATA/fold.csv
  data_path: DATA/resize/norm/
  val_fold: 1
  batch_size: 1
  num_workers: 8
  manual_seed: 2020
  is_segmentation: true
  use_keypoint: True
  dist_thr: 4
  atlas: True
  gaussian: false