BiliSakura's picture
Upload SatMAE++ transformers checkpoints with model card metadata
f18109c verified
Raw
History Blame Contribute Delete
1.68 kB
{
"architectures": [
"SatMAEppModel"
],
"attention_probs_dropout_prob": 0.0,
"channel_embed_dim": 256,
"channel_groups": [
[
0,
1,
2,
6
],
[
3,
4,
5,
7
],
[
8,
9
]
],
"channel_order": "rgb",
"checkpoint_stage": "pretrain",
"dataset": "fmow_sentinel",
"dtype": "float32",
"encoder_type": "group_channel",
"global_pool": true,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"id2label": {},
"image_mean": [
1184.3824625,
1120.77120066,
1136.26026392,
1263.73947144,
1645.40315151,
1846.87040806,
1762.59530783,
1972.62420416,
1732.16362238,
1247.91870117
],
"image_size": 96,
"image_std": [
650.2842772,
712.12507725,
965.23119807,
948.9819932,
1108.06650639,
1258.36394548,
1233.1492281,
1364.38688993,
1310.36996126,
1087.6020813
],
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {},
"layer_norm_eps": 1e-06,
"mlp_ratio": 4.0,
"model_type": "satmae_pp",
"num_attention_heads": 16,
"num_channels": 10,
"num_hidden_layers": 24,
"patch_size": 8,
"qkv_bias": true,
"transformers_version": "5.0.0",
"auto_map": {
"AutoConfig": "modeling_satmae_pp.SatMAEppConfig",
"AutoModel": "modeling_satmae_pp.SatMAEppModel",
"AutoModelForImageClassification": "modeling_satmae_pp.SatMAEppForImageClassification"
},
"custom_pipelines": {
"satmae-pp-feature-extraction": {
"impl": "pipeline_satmae_pp.SatMAEppImageFeatureExtractionPipeline",
"pt": [
"AutoModel"
]
}
}
}