TCF/config.json

{
  "additional_vector_chans": 0,
  "aggregation_method": "custom_attention",
  "attn_drop_rate": 0.1,
  "blur_sigma": null,
  "decoder_type": "trilinear",
  "depth": [
    1,
    2,
    4,
    2,
    1
  ],
  "drop_path_rate": 0.1,
  "drop_rate": 0.1,
  "embed_dim": 48,
  "embed_type": "AOVC",
  "euler_connection": true,
  "img_D": 96,
  "img_H": 96,
  "img_W": 96,
  "in_channels": 4,
  "intra_patch_pos_encoding": false,
  "local_attention": false,
  "max_embed_dim": 192,
  "mlp_ratio": 4.0,
  "num_heads": 8,
  "out_channels": 4,
  "patch_merge": "global",
  "patch_size": 3,
  "periodic": true,
  "pos_encoding_type": "eq_rel_simple",
  "primary_vector_chans": 3,
  "primary_vector_start_idx": 0,
  "scalar_chans": 1,
  "window_size": 2
}