CineMA / pretrained /config.yaml
mathpluscode's picture
Move files
e0466ed
raw
history blame contribute delete
348 Bytes
grad_ckpt: True
data:
sax:
spacing: [1.0, 1.0, 10.0]
patch_size: [192, 192, 16]
in_chans: 1
lax:
spacing: [1.0, 1.0]
patch_size: [256, 256]
in_chans: 1
model:
size: base
views: ["sax", "lax_2c", "lax_3c", "lax_4c"]
patch_size: [4, 4, 1]
scale_factor: [2, 2, 1]
enc_conv_chans: [64, 128]
enc_conv_n_blocks: 2