File size: 435 Bytes
873e8c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
grad_ckpt: True
data:
sax:
spacing: [1.0, 1.0, 10.0]
patch_size: [192, 192, 16]
in_chans: 1
class_column: pathology
pathology: ["ARR", "CIA", "FALL", "HCM", "LV", "NOR"]
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
|