config.json

{
  "architectures": [
    "MyModel"
  ],
  "checkpoint": "mae_in1k",
  "decoder_channels": [
    512,
    256,
    128
  ],
  "final_decoder_channels": 64,
  "hiera_checkpoint": "mae_in1k",
  "hiera_s1_ch": 144,
  "hiera_s2_ch": 288,
  "hiera_s3_ch": 576,
  "hiera_s4_ch": 1152,
  "in_chans": 11,
  "mask_ratio": 0.4,
  "model_type": "my_unet_with_hiera",
  "num_classes": 2,
  "pretrained": true,
  "pretrained_hiera": false,
  "torch_dtype": "float32",
  "transformers_version": "4.51.3"
}