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