grad_ckpt: True data: lax: spacing: [1.0, 1.0] patch_size: [256, 256] in_chans: 1 class_column: pathology pathology: ["ARR", "CIA", "FALL", "HCM", "LV", "NOR"] model: name: convvit views: lax_4c n_frames: 2 out_chans: null convvit: size: base enc_patch_size: [4, 4, 1] enc_scale_factor: [2, 2, 1] enc_conv_chans: [64, 128] enc_conv_n_blocks: 2 dropout: 0.1 drop_path: 0.1