Add hidden_sizes config attribute

#1
by adirik - opened
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -8,6 +8,12 @@
8
  18,
9
  2
10
  ],
 
 
 
 
 
 
11
  "drop_path_rate": 0.1,
12
  "embed_dim": 96,
13
  "encoder_stride": 32,
 
8
  18,
9
  2
10
  ],
11
+ "hidden_sizes": [
12
+ 192,
13
+ 384,
14
+ 768,
15
+ 768
16
+ ],
17
  "drop_path_rate": 0.1,
18
  "embed_dim": 96,
19
  "encoder_stride": 32,