Ændra Rininsland commited on
Commit
a7cdd70
1 Parent(s): e921fc8
config.json CHANGED
@@ -1,89 +1,88 @@
1
- {
2
- "_name_or_path": "microsoft/swinv2-small-patch4-window16-256",
3
- "architectures": [
4
- "Swinv2ForImageClassification"
5
- ],
6
- "attention_probs_dropout_prob": 0.0,
7
- "depths": [
8
- 2,
9
- 2,
10
- 18,
11
- 2
12
- ],
13
- "drop_path_rate": 0.1,
14
- "embed_dim": 96,
15
- "encoder_stride": 32,
16
- "hidden_act": "gelu",
17
- "hidden_dropout_prob": 0.0,
18
- "hidden_size": 768,
19
- "id2label": {
20
- "0": "altright",
21
- "1": "bluesky",
22
- "2": "discord",
23
- "3": "facebook",
24
- "4": "fediverse",
25
- "5": "instagram",
26
- "6": "negative",
27
- "7": "news",
28
- "8": "ngl",
29
- "9": "reddit",
30
- "10": "threads",
31
- "11": "tumblr",
32
- "12": "twitter"
33
- },
34
- "image_size": 256,
35
- "initializer_range": 0.02,
36
- "label2id": {
37
- "altright": 0,
38
- "bluesky": 1,
39
- "discord": 2,
40
- "facebook": 3,
41
- "fediverse": 4,
42
- "instagram": 5,
43
- "negative": 6,
44
- "news": 7,
45
- "ngl": 8,
46
- "reddit": 9,
47
- "threads": 10,
48
- "tumblr": 11,
49
- "twitter": 12
50
- },
51
- "layer_norm_eps": 1e-05,
52
- "mlp_ratio": 4.0,
53
- "model_type": "swinv2",
54
- "num_channels": 3,
55
- "num_heads": [
56
- 3,
57
- 6,
58
- 12,
59
- 24
60
- ],
61
- "num_layers": 4,
62
- "out_features": [
63
- "stage4"
64
- ],
65
- "out_indices": [
66
- 4
67
- ],
68
- "patch_size": 4,
69
- "path_norm": true,
70
- "pretrained_window_sizes": [
71
- 0,
72
- 0,
73
- 0,
74
- 0
75
- ],
76
- "problem_type": "multi_label_classification",
77
- "qkv_bias": true,
78
- "stage_names": [
79
- "stem",
80
- "stage1",
81
- "stage2",
82
- "stage3",
83
- "stage4"
84
- ],
85
- "torch_dtype": "float32",
86
- "transformers_version": "4.44.1",
87
- "use_absolute_embeddings": false,
88
- "window_size": 16
89
- }
 
1
+ {
2
+ "_name_or_path": "howdyaendra/microsoft-swinv2-small-patch4-window16-256-finetuned-xblockm",
3
+ "architectures": [
4
+ "Swinv2ForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 18,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 96,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 768,
19
+ "id2label": {
20
+ "0": "altright",
21
+ "1": "bluesky",
22
+ "2": "discord",
23
+ "3": "facebook",
24
+ "4": "fediverse",
25
+ "5": "instagram",
26
+ "6": "negative",
27
+ "7": "news",
28
+ "8": "ngl",
29
+ "9": "reddit",
30
+ "10": "threads",
31
+ "11": "tumblr",
32
+ "12": "twitter"
33
+ },
34
+ "image_size": 256,
35
+ "initializer_range": 0.02,
36
+ "label2id": {
37
+ "altright": 0,
38
+ "bluesky": 1,
39
+ "discord": 2,
40
+ "facebook": 3,
41
+ "fediverse": 4,
42
+ "instagram": 5,
43
+ "negative": 6,
44
+ "news": 7,
45
+ "ngl": 8,
46
+ "reddit": 9,
47
+ "threads": 10,
48
+ "tumblr": 11,
49
+ "twitter": 12
50
+ },
51
+ "layer_norm_eps": 1e-05,
52
+ "mlp_ratio": 4.0,
53
+ "model_type": "swinv2",
54
+ "num_channels": 3,
55
+ "num_heads": [
56
+ 3,
57
+ 6,
58
+ 12,
59
+ 24
60
+ ],
61
+ "num_layers": 4,
62
+ "out_features": [
63
+ "stage4"
64
+ ],
65
+ "out_indices": [
66
+ 4
67
+ ],
68
+ "patch_size": 4,
69
+ "path_norm": true,
70
+ "pretrained_window_sizes": [
71
+ 0,
72
+ 0,
73
+ 0,
74
+ 0
75
+ ],
76
+ "problem_type": "multi_label_classification",
77
+ "qkv_bias": true,
78
+ "stage_names": [
79
+ "stem",
80
+ "stage1",
81
+ "stage2",
82
+ "stage3",
83
+ "stage4"
84
+ ],
85
+ "transformers_version": "4.33.2",
86
+ "use_absolute_embeddings": false,
87
+ "window_size": 16
88
+ }
 
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2bfce23133841b1282322ccbb116cceddc01cf5e0955c9100c598af6d9fd0e3
3
+ size 198407457
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9dabe3dad2c66d56f4946ad96d818c85721a045d25e3f28019c9ccfadb11ab3
3
+ size 53311995
quantize_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_channel": true,
3
+ "reduce_range": true,
4
+ "per_model_config": {
5
+ "model": {
6
+ "op_types": [
7
+ "Clip",
8
+ "Div",
9
+ "Shape",
10
+ "Mul",
11
+ "Add",
12
+ "Reshape",
13
+ "ReduceMean",
14
+ "ReduceL2",
15
+ "ConstantOfShape",
16
+ "Pad",
17
+ "ScatterND",
18
+ "Range",
19
+ "Concat",
20
+ "Where",
21
+ "Exp",
22
+ "Flatten",
23
+ "Softmax",
24
+ "Sqrt",
25
+ "Unsqueeze",
26
+ "Cast",
27
+ "Not",
28
+ "Sub",
29
+ "MatMul",
30
+ "Sigmoid",
31
+ "Relu",
32
+ "Expand",
33
+ "Conv",
34
+ "Erf",
35
+ "Gemm",
36
+ "GlobalAveragePool",
37
+ "Equal",
38
+ "Transpose",
39
+ "Constant",
40
+ "Slice",
41
+ "Gather",
42
+ "Mod",
43
+ "Pow"
44
+ ],
45
+ "weight_type": "QUInt8"
46
+ }
47
+ }
48
+ }