_base_: 'mri_spine_seg_1e-1_big_rmresizecrop_class20.yml'

model:
  type: VNetDeepSup
  elu: False
  in_channels: 1
  num_classes: 20
  pretrained: null
  kernel_size: [[2,2,4], [2,2,2], [2,2,2], [2,2,2]]
  stride_size: [[2,2,1], [2,2,1], [2,2,2], [2,2,2]]

loss:
  types:
    - type: MixedLoss
      losses:
        - type: CrossEntropyLoss
          weight: Null
        - type: DiceLoss
      coef: [1, 1]
  coef: [0.25, 0.25, 0.25, 0.25]
