Commit
•
12556f8
1
Parent(s):
219253e
Upload model
Browse files- config.json +25 -15
config.json
CHANGED
@@ -1,13 +1,10 @@
|
|
1 |
{
|
2 |
-
"_out_features": [
|
3 |
-
"stage4"
|
4 |
-
],
|
5 |
-
"_out_indices": [
|
6 |
-
4
|
7 |
-
],
|
8 |
"architectures": [
|
9 |
"HieraModel"
|
10 |
],
|
|
|
|
|
|
|
11 |
"depths": [
|
12 |
1,
|
13 |
2,
|
@@ -19,14 +16,14 @@
|
|
19 |
"embed_dim_multiplier": 2.0,
|
20 |
"hidden_act": "gelu",
|
21 |
"hidden_size": 768,
|
22 |
-
"
|
23 |
-
"initializer_range": 0.02,
|
24 |
-
"input_size": [
|
25 |
224,
|
26 |
224
|
27 |
],
|
|
|
28 |
"layer_norm_eps": 1e-06,
|
29 |
"layer_norm_init": 1.0,
|
|
|
30 |
"masked_unit_attention": [
|
31 |
true,
|
32 |
true,
|
@@ -39,17 +36,30 @@
|
|
39 |
],
|
40 |
"mlp_ratio": 4.0,
|
41 |
"model_type": "hiera",
|
|
|
42 |
"num_channels": 3,
|
43 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
44 |
"num_query_pool": 3,
|
45 |
-
"
|
46 |
-
|
47 |
-
|
|
|
|
|
48 |
],
|
49 |
"patch_padding": [
|
50 |
3,
|
51 |
3
|
52 |
],
|
|
|
|
|
|
|
|
|
53 |
"patch_stride": [
|
54 |
4,
|
55 |
4
|
@@ -58,7 +68,6 @@
|
|
58 |
2,
|
59 |
2
|
60 |
],
|
61 |
-
"sep_pos_embed": false,
|
62 |
"stage_names": [
|
63 |
"stem",
|
64 |
"stage1",
|
@@ -67,5 +76,6 @@
|
|
67 |
"stage4"
|
68 |
],
|
69 |
"torch_dtype": "float32",
|
70 |
-
"transformers_version": "4.
|
|
|
71 |
}
|
|
|
1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
"architectures": [
|
3 |
"HieraModel"
|
4 |
],
|
5 |
+
"decoder_depth": null,
|
6 |
+
"decoder_hidden_size": null,
|
7 |
+
"decoder_num_heads": null,
|
8 |
"depths": [
|
9 |
1,
|
10 |
2,
|
|
|
16 |
"embed_dim_multiplier": 2.0,
|
17 |
"hidden_act": "gelu",
|
18 |
"hidden_size": 768,
|
19 |
+
"image_size": [
|
|
|
|
|
20 |
224,
|
21 |
224
|
22 |
],
|
23 |
+
"initializer_range": 0.02,
|
24 |
"layer_norm_eps": 1e-06,
|
25 |
"layer_norm_init": 1.0,
|
26 |
+
"mask_ratio": 0.6,
|
27 |
"masked_unit_attention": [
|
28 |
true,
|
29 |
true,
|
|
|
36 |
],
|
37 |
"mlp_ratio": 4.0,
|
38 |
"model_type": "hiera",
|
39 |
+
"norm_pix_loss": true,
|
40 |
"num_channels": 3,
|
41 |
+
"num_heads": [
|
42 |
+
1,
|
43 |
+
2,
|
44 |
+
4,
|
45 |
+
8
|
46 |
+
],
|
47 |
+
"num_layers": 4,
|
48 |
"num_query_pool": 3,
|
49 |
+
"out_features": [
|
50 |
+
"stage4"
|
51 |
+
],
|
52 |
+
"out_indices": [
|
53 |
+
4
|
54 |
],
|
55 |
"patch_padding": [
|
56 |
3,
|
57 |
3
|
58 |
],
|
59 |
+
"patch_size": [
|
60 |
+
7,
|
61 |
+
7
|
62 |
+
],
|
63 |
"patch_stride": [
|
64 |
4,
|
65 |
4
|
|
|
68 |
2,
|
69 |
2
|
70 |
],
|
|
|
71 |
"stage_names": [
|
72 |
"stem",
|
73 |
"stage1",
|
|
|
76 |
"stage4"
|
77 |
],
|
78 |
"torch_dtype": "float32",
|
79 |
+
"transformers_version": "4.41.0.dev0",
|
80 |
+
"use_separate_position_embedding": false
|
81 |
}
|