HTill's picture
Update config.json
1b276b7 verified
raw
history blame contribute delete
845 Bytes
{
"architectures": [
"EATModel"
],
"auto_map": {
"AutoModel": "modeling_eat.EATModel",
"AutoConfig": "configuration_eat.EATConfig"
},
"model_type": "eat",
"model_variant": "pretrain",
"_comment": "--- STRUCTURAL (DO NOT TOUCH) ---",
"embed_dim": 768,
"depth": 12,
"num_heads": 12,
"patch_size": 16,
"stride": 16,
"in_chans": 1,
"num_classes": 527,
"mlp_ratio": 4.0,
"qkv_bias": true,
"layer_norm_first": false,
"norm_affine": true,
"norm_eps": 1e-06,
"fixed_positions": true,
"_comment": "--- FINE-TUNING KNOBS ---",
"drop_rate": 0.0,
"attn_drop_rate": 0.0,
"post_mlp_drop": 0.0,
"start_drop_path_rate": 0.0,
"end_drop_path_rate": 0.0,
"activation_dropout": 0.0,
"torch_dtype": "float32",
"transformers_version": "4.51.3"
}