grad_ckpt: True data: lax: spacing: [1.0, 1.0] patch_size: [256, 256] in_chans: 1 class_column: vendor vendor: ["GE MEDICAL SYSTEMS", "Philips Medical Systems", "SIEMENS"] 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