config.json

{
  "architectures": [
    "AgriFMModel"
  ],
  "dtype": "float32",
  "embed_dim": 1024,
  "encoder_cfg": {
    "HLSL30": {
      "attn_drop_rate": 0.0,
      "depths": [
        2,
        2,
        18,
        2
      ],
      "downsample_steps": [
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ]
      ],
      "drop_path_rate": 0.2,
      "drop_rate": 0.0,
      "feature_fusion": "cat",
      "frozen_stages": -1,
      "in_chans": 6,
      "mean_frame_down": true,
      "mlp_ratio": 4.0,
      "num_heads": [
        4,
        8,
        16,
        32
      ],
      "out_indices": [
        0,
        1,
        2,
        3
      ],
      "patch_norm": false,
      "patch_size": [
        4,
        4,
        4
      ],
      "qk_scale": null,
      "qkv_bias": true,
      "swin_embed_dim": 128,
      "use_checkpoint": false,
      "window_size": [
        8,
        7,
        7
      ]
    },
    "Modis": {
      "attn_drop_rate": 0.0,
      "depths": [
        2,
        2,
        18,
        2
      ],
      "downsample_steps": [
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ]
      ],
      "drop_path_rate": 0.2,
      "drop_rate": 0.0,
      "feature_fusion": "cat",
      "frozen_stages": -1,
      "in_chans": 7,
      "mean_frame_down": true,
      "mlp_ratio": 4.0,
      "num_heads": [
        4,
        8,
        16,
        32
      ],
      "out_indices": [
        0,
        1,
        2,
        3
      ],
      "patch_norm": false,
      "patch_size": [
        4,
        4,
        4
      ],
      "qk_scale": null,
      "qkv_bias": true,
      "swin_embed_dim": 128,
      "use_checkpoint": false,
      "window_size": [
        8,
        7,
        7
      ]
    },
    "S2": {
      "attn_drop_rate": 0.0,
      "depths": [
        2,
        2,
        18,
        2
      ],
      "downsample_steps": [
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ],
        [
          2,
          2,
          2
        ]
      ],
      "drop_path_rate": 0.2,
      "drop_rate": 0.0,
      "feature_fusion": "cat",
      "frozen_stages": -1,
      "in_chans": 10,
      "mean_frame_down": true,
      "mlp_ratio": 4.0,
      "num_heads": [
        4,
        8,
        16,
        32
      ],
      "out_indices": [
        0,
        1,
        2,
        3
      ],
      "patch_norm": false,
      "patch_size": [
        4,
        4,
        4
      ],
      "qk_scale": null,
      "qkv_bias": true,
      "swin_embed_dim": 128,
      "use_checkpoint": false,
      "window_size": [
        8,
        7,
        7
      ]
    }
  },
  "img_size": 256,
  "loss_ignore_index": -1,
  "model_type": "agrifm",
  "neck_feature_size": [
    16,
    16
  ],
  "neck_fusion_key_list": [
    {
      "HLSL30": 1024,
      "Modis": 1024,
      "S2": 1024
    },
    {
      "HLSL30": 1024,
      "Modis": 1024,
      "S2": 1024
    },
    {
      "HLSL30": 1024,
      "Modis": 1024,
      "S2": 1024
    }
  ],
  "neck_in_feature_key": [
    "HLSL30",
    "S2",
    "Modis"
  ],
  "neck_out_size": [
    256,
    256
  ],
  "num_classes": 2,
  "num_frames": 32,
  "pretrained_weights_path": null,
  "transformers_version": "5.0.0"
}