sayeed99 commited on
Commit
d0ead9b
1 Parent(s): ac8510d

End of training

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. config.json +188 -0
  3. pytorch_model.bin +3 -0
  4. training_args.bin +3 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
config.json ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b2",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 4,
12
+ 6,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "id2label": {
31
+ "0": "Background",
32
+ "1": "shirt, blouse",
33
+ "2": "top, t-shirt, sweatshirt",
34
+ "3": "sweater",
35
+ "4": "cardigan",
36
+ "5": "jacket",
37
+ "6": "vest",
38
+ "7": "pants",
39
+ "8": "shorts",
40
+ "9": "skirt",
41
+ "10": "coat",
42
+ "11": "dress",
43
+ "12": "jumpsuit",
44
+ "13": "cape",
45
+ "14": "glasses",
46
+ "15": "hat",
47
+ "16": "headband, head covering, hair accessory",
48
+ "17": "tie",
49
+ "18": "glove",
50
+ "19": "watch",
51
+ "20": "belt",
52
+ "21": "leg warmer",
53
+ "22": "tights, stockings",
54
+ "23": "sock",
55
+ "24": "shoe",
56
+ "25": "bag, wallet",
57
+ "26": "scarf",
58
+ "27": "umbrella",
59
+ "28": "hood",
60
+ "29": "collar",
61
+ "30": "lapel",
62
+ "31": "epaulette",
63
+ "32": "sleeve",
64
+ "33": "pocket",
65
+ "34": "neckline",
66
+ "35": "buckle",
67
+ "36": "zipper",
68
+ "37": "applique",
69
+ "38": "bead",
70
+ "39": "bow",
71
+ "40": "flower",
72
+ "41": "fringe",
73
+ "42": "ribbon",
74
+ "43": "rivet",
75
+ "44": "ruffle",
76
+ "45": "sequin",
77
+ "46": "tassel",
78
+ "47": "Hair",
79
+ "48": "Sunglasses",
80
+ "49": "Upper-clothes",
81
+ "50": "Left-shoe",
82
+ "51": "Right-shoe",
83
+ "52": "Face",
84
+ "53": "Left-leg",
85
+ "54": "Right-leg",
86
+ "55": "Left-arm",
87
+ "56": "Right-arm"
88
+ },
89
+ "image_size": 224,
90
+ "initializer_range": 0.02,
91
+ "label2id": {
92
+ "Background": 0,
93
+ "Face": 52,
94
+ "Hair": 47,
95
+ "Left-arm": 55,
96
+ "Left-leg": 53,
97
+ "Left-shoe": 50,
98
+ "Right-arm": 56,
99
+ "Right-leg": 54,
100
+ "Right-shoe": 51,
101
+ "Sunglasses": 48,
102
+ "Upper-clothes": 49,
103
+ "applique": 37,
104
+ "bag, wallet": 25,
105
+ "bead": 38,
106
+ "belt": 20,
107
+ "bow": 39,
108
+ "buckle": 35,
109
+ "cape": 13,
110
+ "cardigan": 4,
111
+ "coat": 10,
112
+ "collar": 29,
113
+ "dress": 11,
114
+ "epaulette": 31,
115
+ "flower": 40,
116
+ "fringe": 41,
117
+ "glasses": 14,
118
+ "glove": 18,
119
+ "hat": 15,
120
+ "headband, head covering, hair accessory": 16,
121
+ "hood": 28,
122
+ "jacket": 5,
123
+ "jumpsuit": 12,
124
+ "lapel": 30,
125
+ "leg warmer": 21,
126
+ "neckline": 34,
127
+ "pants": 7,
128
+ "pocket": 33,
129
+ "ribbon": 42,
130
+ "rivet": 43,
131
+ "ruffle": 44,
132
+ "scarf": 26,
133
+ "sequin": 45,
134
+ "shirt, blouse": 1,
135
+ "shoe": 24,
136
+ "shorts": 8,
137
+ "skirt": 9,
138
+ "sleeve": 32,
139
+ "sock": 23,
140
+ "sweater": 3,
141
+ "tassel": 46,
142
+ "tie": 17,
143
+ "tights, stockings": 22,
144
+ "top, t-shirt, sweatshirt": 2,
145
+ "umbrella": 27,
146
+ "vest": 6,
147
+ "watch": 19,
148
+ "zipper": 36
149
+ },
150
+ "layer_norm_eps": 1e-06,
151
+ "mlp_ratios": [
152
+ 4,
153
+ 4,
154
+ 4,
155
+ 4
156
+ ],
157
+ "model_type": "segformer",
158
+ "num_attention_heads": [
159
+ 1,
160
+ 2,
161
+ 5,
162
+ 8
163
+ ],
164
+ "num_channels": 3,
165
+ "num_encoder_blocks": 4,
166
+ "patch_sizes": [
167
+ 7,
168
+ 3,
169
+ 3,
170
+ 3
171
+ ],
172
+ "reshape_last_stage": true,
173
+ "semantic_loss_ignore_index": 255,
174
+ "sr_ratios": [
175
+ 8,
176
+ 4,
177
+ 2,
178
+ 1
179
+ ],
180
+ "strides": [
181
+ 4,
182
+ 2,
183
+ 2,
184
+ 2
185
+ ],
186
+ "torch_dtype": "float32",
187
+ "transformers_version": "4.30.0"
188
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36d4e5e58b7edfd989a9c4bb892bbf45305b725db0f24b157ee85801c14634a1
3
+ size 109699386
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcb44734e5a12d2d68a25328a7d3267b8fb908381d580159d509167b5c29d98a
3
+ size 4408