grad_ckpt: True data: lax: spacing: [1.0, 1.0] patch_size: [256, 256] in_chans: 1 model: name: convunetr views: lax_4c out_chans: 4 convunetr: size: base enc_patch_size: [4, 4, 1] enc_scale_factor: [2, 2, 1] enc_conv_chans: [64, 128] enc_conv_n_blocks: 2 dec_chans: [32, 64, 128, 256, 512] dec_patch_size: [2, 2, 1] dec_scale_factor: [2, 2, 1] dropout: 0.1 drop_path: 0.1