| grad_ckpt: True | |
| data: | |
| sax: | |
| spacing: [1.0, 1.0, 10.0] | |
| patch_size: [192, 192, 16] | |
| in_chans: 1 | |
| class_column: pathology | |
| pathology: ["DCM", "HCM", "NOR", "ARV", "HHD"] | |
| model: | |
| name: convvit | |
| views: sax | |
| 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 | |