Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "activation_dropout": 0.0,
+  
   "architectures": [
     "EATModel"
   ],
@@ -7,26 +7,33 @@
     "AutoModel": "modeling_eat.EATModel",
     "AutoConfig": "configuration_eat.EATConfig"
   },
-  "attn_drop_rate": 0.0,
-  "depth": 12,
-  "drop_rate": 0.0,
-  "embed_dim": 768,
-  "end_drop_path_rate": 0.0,
-  "fixed_positions": true,
-  "in_chans": 1,
-  "layer_norm_first": false,
-  "mlp_ratio": 4.0,
   "model_type": "eat",
   "model_variant": "pretrain",
-  "norm_affine": true,
-  "norm_eps": 1e-06,
-  "num_classes": 527,
-  "num_heads": 12,
-  "patch_size": 16,
-  "post_mlp_drop": 0.0,
-  "qkv_bias": true,
-  "start_drop_path_rate": 0.0,
-  "stride": 16,
+
+  "_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"
 }