ksmcg commited on
Commit
926cae4
1 Parent(s): d4e2bfd

Upload config

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -6,12 +6,12 @@
6
  "channel_dims": null,
7
  "channels": 256,
8
  "cls_attn_layers": 2,
 
9
  "decoder_hidden_size": 768,
10
  "depth": 12,
11
  "depths": null,
12
  "drop_path_rate": 0.0,
13
  "drop_rate": 0.0,
14
- "dropout_ratio": 0.1,
15
  "embed_dim": 384,
16
  "eta": 1.0,
17
  "feat_downsample": false,
@@ -46,6 +46,7 @@
46
  "num_attention_heads": 8,
47
  "num_classes": 1000,
48
  "num_heads": 8,
 
49
  "out_index": -1,
50
  "patch_size": 16,
51
  "qkv_bias": true,
 
6
  "channel_dims": null,
7
  "channels": 256,
8
  "cls_attn_layers": 2,
9
+ "decoder_dropout": 0.1,
10
  "decoder_hidden_size": 768,
11
  "depth": 12,
12
  "depths": null,
13
  "drop_path_rate": 0.0,
14
  "drop_rate": 0.0,
 
15
  "embed_dim": 384,
16
  "eta": 1.0,
17
  "feat_downsample": false,
 
46
  "num_attention_heads": 8,
47
  "num_classes": 1000,
48
  "num_heads": 8,
49
+ "num_hidden_layers": 12,
50
  "out_index": -1,
51
  "patch_size": 16,
52
  "qkv_bias": true,