sdoerrich97
/

stylizing_vit_tiny_epistr

domain-generalization

vision-transformer

Model card Files Files and versions

sdoerrich97 commited on 24 days ago

Commit

f7aafa9

·

verified ·

1 Parent(s): 51dacee

Update config.json

Files changed (1) hide show

config.json +6 -5

config.json CHANGED Viewed

@@ -4,12 +4,13 @@
   ],
   "model_type": "stylizing_vit",
   "backbone": "tiny",
-  "hidden_size": 192,
-  "num_hidden_layers": 12,
-  "num_attention_heads": 3,
-  "intermediate_size": 768,
-  "image_size": 224,
   "patch_size": 16,
   "mlp_ratio": 4.0,
   "layer_norm_eps": 1e-6
 }

   ],
   "model_type": "stylizing_vit",
   "backbone": "tiny",
+  "input_size": 224,
+  "in_chans": 3,
   "patch_size": 16,
+  "embed_dim": 192,
+  "depth": 12,
+  "num_heads": 3,
   "mlp_ratio": 4.0,
+  "qkv_bias": true,
   "layer_norm_eps": 1e-6
 }