ksmcg commited on
Commit
4ad8e13
1 Parent(s): 494476e

Upload config

Browse files
Files changed (1) hide show
  1. config.json +2 -0
config.json CHANGED
@@ -17,6 +17,7 @@
17
  ],
18
  "channels": 256,
19
  "cls_attn_layers": 2,
 
20
  "decoder_hidden_size": 768,
21
  "decoder_params": {
22
  "embed_dim": 768
@@ -69,6 +70,7 @@
69
  "num_attention_heads": 8,
70
  "num_classes": 1000,
71
  "num_heads": 8,
 
72
  "out_index": 9,
73
  "patch_size": 16,
74
  "qkv_bias": true,
 
17
  ],
18
  "channels": 256,
19
  "cls_attn_layers": 2,
20
+ "decoder_dropout": 0.1,
21
  "decoder_hidden_size": 768,
22
  "decoder_params": {
23
  "embed_dim": 768
 
70
  "num_attention_heads": 8,
71
  "num_classes": 1000,
72
  "num_heads": 8,
73
+ "num_hidden_layers": 12,
74
  "out_index": 9,
75
  "patch_size": 16,
76
  "qkv_bias": true,