grad_ckpt: True
data:
  lax:
    spacing: [1.0, 1.0]
    patch_size: [256, 256]
    in_chans: 1
model:
  name: convvit
  views: "lax_2c"
  n_frames: 1
  out_chans: 6
  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