Hugging Face's logo

chiHang
/
detr-resnet-50_finetuned_cppe5

Object Detection
Transformers
PyTorch
TensorBoard
detr
Model card Files
xet
Metrics Community
1
chiHang commited on
Commit
58aa8ef
·
1 Parent(s): 2ffc2cb

Upload 36 files

Browse files
Files changed (36) hide show
  1. checkpoint-1800/config.json +55 -0
  2. checkpoint-1800/optimizer.pt +3 -0
  3. checkpoint-1800/preprocessor_config.json +24 -0
  4. checkpoint-1800/pytorch_model.bin +3 -0
  5. checkpoint-1800/rng_state.pth +3 -0
  6. checkpoint-1800/scaler.pt +3 -0
  7. checkpoint-1800/scheduler.pt +3 -0
  8. checkpoint-1800/trainer_state.json +232 -0
  9. checkpoint-1800/training_args.bin +3 -0
  10. checkpoint-2000/config.json +55 -0
  11. checkpoint-2000/optimizer.pt +3 -0
  12. checkpoint-2000/preprocessor_config.json +24 -0
  13. checkpoint-2000/pytorch_model.bin +3 -0
  14. checkpoint-2000/rng_state.pth +3 -0
  15. checkpoint-2000/scaler.pt +3 -0
  16. checkpoint-2000/scheduler.pt +3 -0
  17. checkpoint-2000/trainer_state.json +256 -0
  18. checkpoint-2000/training_args.bin +3 -0
  19. config.json +55 -0
  20. preprocessor_config.json +24 -0
  21. pytorch_model.bin +3 -0
  22. runs/Feb02_10-06-07_2e45d81e608c/1675332492.6842287/events.out.tfevents.1675332492.2e45d81e608c.384.1 +3 -0
  23. runs/Feb02_10-06-07_2e45d81e608c/events.out.tfevents.1675332492.2e45d81e608c.384.0 +3 -0
  24. runs/Feb02_10-09-33_2e45d81e608c/1675332579.8324304/events.out.tfevents.1675332579.2e45d81e608c.384.3 +3 -0
  25. runs/Feb02_10-09-33_2e45d81e608c/events.out.tfevents.1675332579.2e45d81e608c.384.2 +3 -0
  26. runs/Feb02_10-20-32_2e45d81e608c/1675333283.8141575/events.out.tfevents.1675333283.2e45d81e608c.384.7 +3 -0
  27. runs/Feb02_10-20-32_2e45d81e608c/1675333322.9661694/events.out.tfevents.1675333322.2e45d81e608c.384.9 +3 -0
  28. runs/Feb02_10-20-32_2e45d81e608c/1675333362.5032432/events.out.tfevents.1675333362.2e45d81e608c.384.11 +3 -0
  29. runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333283.2e45d81e608c.384.6 +3 -0
  30. runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333322.2e45d81e608c.384.8 +3 -0
  31. runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333362.2e45d81e608c.384.10 +3 -0
  32. runs/Feb02_10-23-25_2e45d81e608c/1675333414.5977795/events.out.tfevents.1675333414.2e45d81e608c.384.13 +3 -0
  33. runs/Feb02_10-23-25_2e45d81e608c/events.out.tfevents.1675333414.2e45d81e608c.384.12 +3 -0
  34. runs/Feb02_16-48-16_6c867e2afd4d/1675356534.8155868/events.out.tfevents.1675356534.6c867e2afd4d.942.3 +3 -0
  35. runs/Feb02_16-48-16_6c867e2afd4d/events.out.tfevents.1675356534.6c867e2afd4d.942.2 +3 -0
  36. training_args.bin +3 -0
checkpoint-1800/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/detr-resnet-50",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "relu",
5
+ "architectures": [
6
+ "DetrForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": false,
10
+ "backbone": "resnet50",
11
+ "backbone_config": null,
12
+ "bbox_cost": 5,
13
+ "bbox_loss_coefficient": 5,
14
+ "class_cost": 1,
15
+ "d_model": 256,
16
+ "decoder_attention_heads": 8,
17
+ "decoder_ffn_dim": 2048,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 6,
20
+ "dice_loss_coefficient": 1,
21
+ "dilation": false,
22
+ "dropout": 0.1,
23
+ "encoder_attention_heads": 8,
24
+ "encoder_ffn_dim": 2048,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 6,
27
+ "eos_coefficient": 0.1,
28
+ "giou_cost": 2,
29
+ "giou_loss_coefficient": 2,
30
+ "id2label": {
31
+ "1": "neck",
32
+ "2": "body",
33
+ "3": "sleeve"
34
+ },
35
+ "init_std": 0.02,
36
+ "init_xavier_std": 1.0,
37
+ "is_encoder_decoder": true,
38
+ "label2id": {
39
+ "body": 2,
40
+ "neck": 1,
41
+ "sleeve": 3
42
+ },
43
+ "mask_loss_coefficient": 1,
44
+ "max_position_embeddings": 1024,
45
+ "model_type": "detr",
46
+ "num_channels": 3,
47
+ "num_hidden_layers": 6,
48
+ "num_queries": 100,
49
+ "position_embedding_type": "sine",
50
+ "scale_embedding": false,
51
+ "torch_dtype": "float32",
52
+ "transformers_version": "4.26.0",
53
+ "use_pretrained_backbone": true,
54
+ "use_timm_backbone": true
55
+ }
checkpoint-1800/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14604ba6a2b698c1e116e121a7003baaae639a46dbdb21d7a56e60ecf6cda3b3
3
+ size 330480261
checkpoint-1800/preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_pad": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "format": "coco_detection",
7
+ "image_mean": [
8
+ 0.485,
9
+ 0.456,
10
+ 0.406
11
+ ],
12
+ "image_processor_type": "DetrImageProcessor",
13
+ "image_std": [
14
+ 0.229,
15
+ 0.224,
16
+ 0.225
17
+ ],
18
+ "resample": 2,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "longest_edge": 1333,
22
+ "shortest_edge": 800
23
+ }
24
+ }
checkpoint-1800/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc72ec4fd72f742d80af08df0a425bfd7527ef4bb11d293248c35237d1d82214
3
+ size 166618149
checkpoint-1800/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1961e772fdc2a568585328a0831d083f26c6dc076c3f04db5dc54352a1223a66
3
+ size 14575
checkpoint-1800/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d0bb9489c5143ec668d0b97718e8568b3135b7aa0b06793bc3f6f97b7b2889
3
+ size 557
checkpoint-1800/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ea4daf6a577546123fe2b6482e3d8c4e97b4f3233958b4f6ddb6811e203cf2c
3
+ size 627
checkpoint-1800/trainer_state.json ADDED
@@ -0,0 +1,232 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 360.0,
5
+ "global_step": 1800,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 10.0,
12
+ "learning_rate": 9.790476190476192e-06,
13
+ "loss": 2.4087,
14
+ "step": 50
15
+ },
16
+ {
17
+ "epoch": 20.0,
18
+ "learning_rate": 9.552380952380953e-06,
19
+ "loss": 1.3645,
20
+ "step": 100
21
+ },
22
+ {
23
+ "epoch": 30.0,
24
+ "learning_rate": 9.314285714285714e-06,
25
+ "loss": 1.0984,
26
+ "step": 150
27
+ },
28
+ {
29
+ "epoch": 40.0,
30
+ "learning_rate": 9.076190476190477e-06,
31
+ "loss": 1.0497,
32
+ "step": 200
33
+ },
34
+ {
35
+ "epoch": 50.0,
36
+ "learning_rate": 8.83809523809524e-06,
37
+ "loss": 0.9478,
38
+ "step": 250
39
+ },
40
+ {
41
+ "epoch": 60.0,
42
+ "learning_rate": 8.6e-06,
43
+ "loss": 0.8743,
44
+ "step": 300
45
+ },
46
+ {
47
+ "epoch": 70.0,
48
+ "learning_rate": 8.361904761904762e-06,
49
+ "loss": 0.8098,
50
+ "step": 350
51
+ },
52
+ {
53
+ "epoch": 80.0,
54
+ "learning_rate": 8.123809523809525e-06,
55
+ "loss": 0.7879,
56
+ "step": 400
57
+ },
58
+ {
59
+ "epoch": 90.0,
60
+ "learning_rate": 7.885714285714286e-06,
61
+ "loss": 0.702,
62
+ "step": 450
63
+ },
64
+ {
65
+ "epoch": 100.0,
66
+ "learning_rate": 7.647619047619049e-06,
67
+ "loss": 0.6197,
68
+ "step": 500
69
+ },
70
+ {
71
+ "epoch": 110.0,
72
+ "learning_rate": 7.40952380952381e-06,
73
+ "loss": 0.5729,
74
+ "step": 550
75
+ },
76
+ {
77
+ "epoch": 120.0,
78
+ "learning_rate": 7.1714285714285725e-06,
79
+ "loss": 0.5353,
80
+ "step": 600
81
+ },
82
+ {
83
+ "epoch": 130.0,
84
+ "learning_rate": 6.9333333333333344e-06,
85
+ "loss": 0.4739,
86
+ "step": 650
87
+ },
88
+ {
89
+ "epoch": 140.0,
90
+ "learning_rate": 6.6952380952380956e-06,
91
+ "loss": 0.4399,
92
+ "step": 700
93
+ },
94
+ {
95
+ "epoch": 150.0,
96
+ "learning_rate": 6.4571428571428575e-06,
97
+ "loss": 0.3892,
98
+ "step": 750
99
+ },
100
+ {
101
+ "epoch": 160.0,
102
+ "learning_rate": 6.21904761904762e-06,
103
+ "loss": 0.3537,
104
+ "step": 800
105
+ },
106
+ {
107
+ "epoch": 170.0,
108
+ "learning_rate": 5.9809523809523814e-06,
109
+ "loss": 0.3312,
110
+ "step": 850
111
+ },
112
+ {
113
+ "epoch": 180.0,
114
+ "learning_rate": 5.742857142857143e-06,
115
+ "loss": 0.3184,
116
+ "step": 900
117
+ },
118
+ {
119
+ "epoch": 190.0,
120
+ "learning_rate": 5.504761904761905e-06,
121
+ "loss": 0.2993,
122
+ "step": 950
123
+ },
124
+ {
125
+ "epoch": 200.0,
126
+ "learning_rate": 5.2666666666666665e-06,
127
+ "loss": 0.2885,
128
+ "step": 1000
129
+ },
130
+ {
131
+ "epoch": 210.0,
132
+ "learning_rate": 5.028571428571429e-06,
133
+ "loss": 0.2745,
134
+ "step": 1050
135
+ },
136
+ {
137
+ "epoch": 220.0,
138
+ "learning_rate": 4.790476190476191e-06,
139
+ "loss": 0.2587,
140
+ "step": 1100
141
+ },
142
+ {
143
+ "epoch": 230.0,
144
+ "learning_rate": 4.552380952380952e-06,
145
+ "loss": 0.2492,
146
+ "step": 1150
147
+ },
148
+ {
149
+ "epoch": 240.0,
150
+ "learning_rate": 4.314285714285714e-06,
151
+ "loss": 0.2436,
152
+ "step": 1200
153
+ },
154
+ {
155
+ "epoch": 250.0,
156
+ "learning_rate": 4.076190476190476e-06,
157
+ "loss": 0.2322,
158
+ "step": 1250
159
+ },
160
+ {
161
+ "epoch": 260.0,
162
+ "learning_rate": 3.838095238095238e-06,
163
+ "loss": 0.234,
164
+ "step": 1300
165
+ },
166
+ {
167
+ "epoch": 270.0,
168
+ "learning_rate": 3.6000000000000003e-06,
169
+ "loss": 0.2289,
170
+ "step": 1350
171
+ },
172
+ {
173
+ "epoch": 280.0,
174
+ "learning_rate": 3.3619047619047622e-06,
175
+ "loss": 0.2175,
176
+ "step": 1400
177
+ },
178
+ {
179
+ "epoch": 290.0,
180
+ "learning_rate": 3.1238095238095238e-06,
181
+ "loss": 0.2128,
182
+ "step": 1450
183
+ },
184
+ {
185
+ "epoch": 300.0,
186
+ "learning_rate": 2.885714285714286e-06,
187
+ "loss": 0.2083,
188
+ "step": 1500
189
+ },
190
+ {
191
+ "epoch": 310.0,
192
+ "learning_rate": 2.6476190476190477e-06,
193
+ "loss": 0.2037,
194
+ "step": 1550
195
+ },
196
+ {
197
+ "epoch": 320.0,
198
+ "learning_rate": 2.4095238095238097e-06,
199
+ "loss": 0.2018,
200
+ "step": 1600
201
+ },
202
+ {
203
+ "epoch": 330.0,
204
+ "learning_rate": 2.1714285714285716e-06,
205
+ "loss": 0.1987,
206
+ "step": 1650
207
+ },
208
+ {
209
+ "epoch": 340.0,
210
+ "learning_rate": 1.9333333333333336e-06,
211
+ "loss": 0.1941,
212
+ "step": 1700
213
+ },
214
+ {
215
+ "epoch": 350.0,
216
+ "learning_rate": 1.6952380952380954e-06,
217
+ "loss": 0.1941,
218
+ "step": 1750
219
+ },
220
+ {
221
+ "epoch": 360.0,
222
+ "learning_rate": 1.4571428571428573e-06,
223
+ "loss": 0.1941,
224
+ "step": 1800
225
+ }
226
+ ],
227
+ "max_steps": 2100,
228
+ "num_train_epochs": 420,
229
+ "total_flos": 6.7096928710656e+18,
230
+ "trial_name": null,
231
+ "trial_params": null
232
+ }
checkpoint-1800/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eb2528745e2a9f7867b19b94f296a3037db4ea8c54ac8d393159b5918e81e51
3
+ size 3643
checkpoint-2000/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/detr-resnet-50",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "relu",
5
+ "architectures": [
6
+ "DetrForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": false,
10
+ "backbone": "resnet50",
11
+ "backbone_config": null,
12
+ "bbox_cost": 5,
13
+ "bbox_loss_coefficient": 5,
14
+ "class_cost": 1,
15
+ "d_model": 256,
16
+ "decoder_attention_heads": 8,
17
+ "decoder_ffn_dim": 2048,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 6,
20
+ "dice_loss_coefficient": 1,
21
+ "dilation": false,
22
+ "dropout": 0.1,
23
+ "encoder_attention_heads": 8,
24
+ "encoder_ffn_dim": 2048,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 6,
27
+ "eos_coefficient": 0.1,
28
+ "giou_cost": 2,
29
+ "giou_loss_coefficient": 2,
30
+ "id2label": {
31
+ "1": "neck",
32
+ "2": "body",
33
+ "3": "sleeve"
34
+ },
35
+ "init_std": 0.02,
36
+ "init_xavier_std": 1.0,
37
+ "is_encoder_decoder": true,
38
+ "label2id": {
39
+ "body": 2,
40
+ "neck": 1,
41
+ "sleeve": 3
42
+ },
43
+ "mask_loss_coefficient": 1,
44
+ "max_position_embeddings": 1024,
45
+ "model_type": "detr",
46
+ "num_channels": 3,
47
+ "num_hidden_layers": 6,
48
+ "num_queries": 100,
49
+ "position_embedding_type": "sine",
50
+ "scale_embedding": false,
51
+ "torch_dtype": "float32",
52
+ "transformers_version": "4.26.0",
53
+ "use_pretrained_backbone": true,
54
+ "use_timm_backbone": true
55
+ }
checkpoint-2000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44583a12df53956ea5db6a6128430ac130d657cfd7e81616697275bbeffb8d7a
3
+ size 330480261
checkpoint-2000/preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_pad": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "format": "coco_detection",
7
+ "image_mean": [
8
+ 0.485,
9
+ 0.456,
10
+ 0.406
11
+ ],
12
+ "image_processor_type": "DetrImageProcessor",
13
+ "image_std": [
14
+ 0.229,
15
+ 0.224,
16
+ 0.225
17
+ ],
18
+ "resample": 2,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "longest_edge": 1333,
22
+ "shortest_edge": 800
23
+ }
24
+ }
checkpoint-2000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0dec7594b7a92e172097837b7225e64feefacb0842e3e088582dbf72fb2975e2
3
+ size 166618149
checkpoint-2000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc146938115c432262cbf69888e661bf1850621c4d048778d8d37b1fdff60b6e
3
+ size 14575
checkpoint-2000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2500405b4dc230cce74b108386b1ff242914fbf7476eb4c7612e8334fdfe2c4f
3
+ size 557
checkpoint-2000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9ef9fbe6489d43be16f69ed5240e1e1598290d621fa52bd7bf9bac7c0f30fec
3
+ size 627
checkpoint-2000/trainer_state.json ADDED
@@ -0,0 +1,256 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 400.0,
5
+ "global_step": 2000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 10.0,
12
+ "learning_rate": 9.790476190476192e-06,
13
+ "loss": 2.4087,
14
+ "step": 50
15
+ },
16
+ {
17
+ "epoch": 20.0,
18
+ "learning_rate": 9.552380952380953e-06,
19
+ "loss": 1.3645,
20
+ "step": 100
21
+ },
22
+ {
23
+ "epoch": 30.0,
24
+ "learning_rate": 9.314285714285714e-06,
25
+ "loss": 1.0984,
26
+ "step": 150
27
+ },
28
+ {
29
+ "epoch": 40.0,
30
+ "learning_rate": 9.076190476190477e-06,
31
+ "loss": 1.0497,
32
+ "step": 200
33
+ },
34
+ {
35
+ "epoch": 50.0,
36
+ "learning_rate": 8.83809523809524e-06,
37
+ "loss": 0.9478,
38
+ "step": 250
39
+ },
40
+ {
41
+ "epoch": 60.0,
42
+ "learning_rate": 8.6e-06,
43
+ "loss": 0.8743,
44
+ "step": 300
45
+ },
46
+ {
47
+ "epoch": 70.0,
48
+ "learning_rate": 8.361904761904762e-06,
49
+ "loss": 0.8098,
50
+ "step": 350
51
+ },
52
+ {
53
+ "epoch": 80.0,
54
+ "learning_rate": 8.123809523809525e-06,
55
+ "loss": 0.7879,
56
+ "step": 400
57
+ },
58
+ {
59
+ "epoch": 90.0,
60
+ "learning_rate": 7.885714285714286e-06,
61
+ "loss": 0.702,
62
+ "step": 450
63
+ },
64
+ {
65
+ "epoch": 100.0,
66
+ "learning_rate": 7.647619047619049e-06,
67
+ "loss": 0.6197,
68
+ "step": 500
69
+ },
70
+ {
71
+ "epoch": 110.0,
72
+ "learning_rate": 7.40952380952381e-06,
73
+ "loss": 0.5729,
74
+ "step": 550
75
+ },
76
+ {
77
+ "epoch": 120.0,
78
+ "learning_rate": 7.1714285714285725e-06,
79
+ "loss": 0.5353,
80
+ "step": 600
81
+ },
82
+ {
83
+ "epoch": 130.0,
84
+ "learning_rate": 6.9333333333333344e-06,
85
+ "loss": 0.4739,
86
+ "step": 650
87
+ },
88
+ {
89
+ "epoch": 140.0,
90
+ "learning_rate": 6.6952380952380956e-06,
91
+ "loss": 0.4399,
92
+ "step": 700
93
+ },
94
+ {
95
+ "epoch": 150.0,
96
+ "learning_rate": 6.4571428571428575e-06,
97
+ "loss": 0.3892,
98
+ "step": 750
99
+ },
100
+ {
101
+ "epoch": 160.0,
102
+ "learning_rate": 6.21904761904762e-06,
103
+ "loss": 0.3537,
104
+ "step": 800
105
+ },
106
+ {
107
+ "epoch": 170.0,
108
+ "learning_rate": 5.9809523809523814e-06,
109
+ "loss": 0.3312,
110
+ "step": 850
111
+ },
112
+ {
113
+ "epoch": 180.0,
114
+ "learning_rate": 5.742857142857143e-06,
115
+ "loss": 0.3184,
116
+ "step": 900
117
+ },
118
+ {
119
+ "epoch": 190.0,
120
+ "learning_rate": 5.504761904761905e-06,
121
+ "loss": 0.2993,
122
+ "step": 950
123
+ },
124
+ {
125
+ "epoch": 200.0,
126
+ "learning_rate": 5.2666666666666665e-06,
127
+ "loss": 0.2885,
128
+ "step": 1000
129
+ },
130
+ {
131
+ "epoch": 210.0,
132
+ "learning_rate": 5.028571428571429e-06,
133
+ "loss": 0.2745,
134
+ "step": 1050
135
+ },
136
+ {
137
+ "epoch": 220.0,
138
+ "learning_rate": 4.790476190476191e-06,
139
+ "loss": 0.2587,
140
+ "step": 1100
141
+ },
142
+ {
143
+ "epoch": 230.0,
144
+ "learning_rate": 4.552380952380952e-06,
145
+ "loss": 0.2492,
146
+ "step": 1150
147
+ },
148
+ {
149
+ "epoch": 240.0,
150
+ "learning_rate": 4.314285714285714e-06,
151
+ "loss": 0.2436,
152
+ "step": 1200
153
+ },
154
+ {
155
+ "epoch": 250.0,
156
+ "learning_rate": 4.076190476190476e-06,
157
+ "loss": 0.2322,
158
+ "step": 1250
159
+ },
160
+ {
161
+ "epoch": 260.0,
162
+ "learning_rate": 3.838095238095238e-06,
163
+ "loss": 0.234,
164
+ "step": 1300
165
+ },
166
+ {
167
+ "epoch": 270.0,
168
+ "learning_rate": 3.6000000000000003e-06,
169
+ "loss": 0.2289,
170
+ "step": 1350
171
+ },
172
+ {
173
+ "epoch": 280.0,
174
+ "learning_rate": 3.3619047619047622e-06,
175
+ "loss": 0.2175,
176
+ "step": 1400
177
+ },
178
+ {
179
+ "epoch": 290.0,
180
+ "learning_rate": 3.1238095238095238e-06,
181
+ "loss": 0.2128,
182
+ "step": 1450
183
+ },
184
+ {
185
+ "epoch": 300.0,
186
+ "learning_rate": 2.885714285714286e-06,
187
+ "loss": 0.2083,
188
+ "step": 1500
189
+ },
190
+ {
191
+ "epoch": 310.0,
192
+ "learning_rate": 2.6476190476190477e-06,
193
+ "loss": 0.2037,
194
+ "step": 1550
195
+ },
196
+ {
197
+ "epoch": 320.0,
198
+ "learning_rate": 2.4095238095238097e-06,
199
+ "loss": 0.2018,
200
+ "step": 1600
201
+ },
202
+ {
203
+ "epoch": 330.0,
204
+ "learning_rate": 2.1714285714285716e-06,
205
+ "loss": 0.1987,
206
+ "step": 1650
207
+ },
208
+ {
209
+ "epoch": 340.0,
210
+ "learning_rate": 1.9333333333333336e-06,
211
+ "loss": 0.1941,
212
+ "step": 1700
213
+ },
214
+ {
215
+ "epoch": 350.0,
216
+ "learning_rate": 1.6952380952380954e-06,
217
+ "loss": 0.1941,
218
+ "step": 1750
219
+ },
220
+ {
221
+ "epoch": 360.0,
222
+ "learning_rate": 1.4571428571428573e-06,
223
+ "loss": 0.1941,
224
+ "step": 1800
225
+ },
226
+ {
227
+ "epoch": 370.0,
228
+ "learning_rate": 1.219047619047619e-06,
229
+ "loss": 0.1892,
230
+ "step": 1850
231
+ },
232
+ {
233
+ "epoch": 380.0,
234
+ "learning_rate": 9.80952380952381e-07,
235
+ "loss": 0.1856,
236
+ "step": 1900
237
+ },
238
+ {
239
+ "epoch": 390.0,
240
+ "learning_rate": 7.428571428571429e-07,
241
+ "loss": 0.183,
242
+ "step": 1950
243
+ },
244
+ {
245
+ "epoch": 400.0,
246
+ "learning_rate": 5.047619047619048e-07,
247
+ "loss": 0.187,
248
+ "step": 2000
249
+ }
250
+ ],
251
+ "max_steps": 2100,
252
+ "num_train_epochs": 420,
253
+ "total_flos": 7.455214301184e+18,
254
+ "trial_name": null,
255
+ "trial_params": null
256
+ }
checkpoint-2000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eb2528745e2a9f7867b19b94f296a3037db4ea8c54ac8d393159b5918e81e51
3
+ size 3643
config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/detr-resnet-50",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "relu",
5
+ "architectures": [
6
+ "DetrForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": false,
10
+ "backbone": "resnet50",
11
+ "backbone_config": null,
12
+ "bbox_cost": 5,
13
+ "bbox_loss_coefficient": 5,
14
+ "class_cost": 1,
15
+ "d_model": 256,
16
+ "decoder_attention_heads": 8,
17
+ "decoder_ffn_dim": 2048,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 6,
20
+ "dice_loss_coefficient": 1,
21
+ "dilation": false,
22
+ "dropout": 0.1,
23
+ "encoder_attention_heads": 8,
24
+ "encoder_ffn_dim": 2048,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 6,
27
+ "eos_coefficient": 0.1,
28
+ "giou_cost": 2,
29
+ "giou_loss_coefficient": 2,
30
+ "id2label": {
31
+ "1": "neck",
32
+ "2": "body",
33
+ "3": "sleeve"
34
+ },
35
+ "init_std": 0.02,
36
+ "init_xavier_std": 1.0,
37
+ "is_encoder_decoder": true,
38
+ "label2id": {
39
+ "body": 2,
40
+ "neck": 1,
41
+ "sleeve": 3
42
+ },
43
+ "mask_loss_coefficient": 1,
44
+ "max_position_embeddings": 1024,
45
+ "model_type": "detr",
46
+ "num_channels": 3,
47
+ "num_hidden_layers": 6,
48
+ "num_queries": 100,
49
+ "position_embedding_type": "sine",
50
+ "scale_embedding": false,
51
+ "torch_dtype": "float32",
52
+ "transformers_version": "4.26.0",
53
+ "use_pretrained_backbone": true,
54
+ "use_timm_backbone": true
55
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_pad": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "format": "coco_detection",
7
+ "image_mean": [
8
+ 0.485,
9
+ 0.456,
10
+ 0.406
11
+ ],
12
+ "image_processor_type": "DetrImageProcessor",
13
+ "image_std": [
14
+ 0.229,
15
+ 0.224,
16
+ 0.225
17
+ ],
18
+ "resample": 2,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "longest_edge": 1333,
22
+ "shortest_edge": 800
23
+ }
24
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89bab988f14850e0d2e5904711a9e682e26688360a30acd4a921e2e452c74eca
3
+ size 166618149
runs/Feb02_10-06-07_2e45d81e608c/1675332492.6842287/events.out.tfevents.1675332492.2e45d81e608c.384.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:795af5e93b64706e78f0a647337d06bd466bc5b4bc89da11cd93d76114d14c33
3
+ size 5707
runs/Feb02_10-06-07_2e45d81e608c/events.out.tfevents.1675332492.2e45d81e608c.384.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a7d6bed4527dadf8a388cc23b93e694e02090c8cb8cce4da5126c92c480a37b
3
+ size 4468
runs/Feb02_10-09-33_2e45d81e608c/1675332579.8324304/events.out.tfevents.1675332579.2e45d81e608c.384.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:579a4bdd819cc58a45035acae571378d7cfab004dba71595457e7d84d7585403
3
+ size 5707
runs/Feb02_10-09-33_2e45d81e608c/events.out.tfevents.1675332579.2e45d81e608c.384.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e71ebb8b14fcaf5c722272057ee4cc20d4e9ffc05ce52c69298e7c8f8733f99
3
+ size 4970
runs/Feb02_10-20-32_2e45d81e608c/1675333283.8141575/events.out.tfevents.1675333283.2e45d81e608c.384.7 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f79d9aaa0459f75ec9ee733503028f7c0a40e01c5ae862736a055399c0ec3dd7
3
+ size 5707
runs/Feb02_10-20-32_2e45d81e608c/1675333322.9661694/events.out.tfevents.1675333322.2e45d81e608c.384.9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d62cddd436ee7c2e321c5760d84edf44c493483d64470e3a11df35c685ba2f1
3
+ size 5707
runs/Feb02_10-20-32_2e45d81e608c/1675333362.5032432/events.out.tfevents.1675333362.2e45d81e608c.384.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e79e2d881c5f389fd69d25717d96034caa93e2cc9fcbe56ebce268f688121f0b
3
+ size 5707
runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333283.2e45d81e608c.384.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40b913ed3f074e747446ac1d91c5a43c06991496871e5d85f2c250da30b94ebc
3
+ size 4161
runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333322.2e45d81e608c.384.8 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c34cda502243124f9996dba01dfda7e6bdc5167276852ce2de10f6eb854a028
3
+ size 4161
runs/Feb02_10-20-32_2e45d81e608c/events.out.tfevents.1675333362.2e45d81e608c.384.10 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d0be9a8ef2ea6b30775d886fb743a85b4b5981d98f43ba53c1083d70196b36f
3
+ size 4161
runs/Feb02_10-23-25_2e45d81e608c/1675333414.5977795/events.out.tfevents.1675333414.2e45d81e608c.384.13 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a947f19f041750111f9168c6ac0819ac00078391ef2faf8e156a03ecf970d7f5
3
+ size 5707
runs/Feb02_10-23-25_2e45d81e608c/events.out.tfevents.1675333414.2e45d81e608c.384.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3c7abe57f55d6cd4026a2b560767549aa44d8c79cdb4c021393bcd59de0de10
3
+ size 11411
runs/Feb02_16-48-16_6c867e2afd4d/1675356534.8155868/events.out.tfevents.1675356534.6c867e2afd4d.942.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ef1cb679d0f3b1189d8de6b1db521d81b32e61caa398ec70b130816a0f48a84
3
+ size 5870
runs/Feb02_16-48-16_6c867e2afd4d/events.out.tfevents.1675356534.6c867e2afd4d.942.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1608f7ae8bd9a4325edc9a124998d92c6f1aa7460c8b4d0887943f35be156811
3
+ size 11573
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eb2528745e2a9f7867b19b94f296a3037db4ea8c54ac8d393159b5918e81e51
3
+ size 3643