qubvel-hf HF staff commited on
Commit
fd80976
1 Parent(s): 978aff3

End of training

Browse files
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: apache-2.0
3
  base_model: SenseTime/deformable-detr
4
  tags:
 
 
5
  - generated_from_trainer
6
  model-index:
7
  - name: sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
14
  [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/qubvel-hf-co/transformers-detection-model-finetuning-cppe5/runs/tmhaicli)
15
  # sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
16
 
17
- This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
  - Loss: 1.0442
20
  - Map: 0.3514
@@ -59,7 +61,7 @@ More information needed
59
  The following hyperparameters were used during training:
60
  - learning_rate: 5e-05
61
  - train_batch_size: 4
62
- - eval_batch_size: 8
63
  - seed: 1337
64
  - gradient_accumulation_steps: 2
65
  - total_train_batch_size: 8
 
2
  license: apache-2.0
3
  base_model: SenseTime/deformable-detr
4
  tags:
5
+ - object-detection
6
+ - vision
7
  - generated_from_trainer
8
  model-index:
9
  - name: sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
 
16
  [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/qubvel-hf-co/transformers-detection-model-finetuning-cppe5/runs/tmhaicli)
17
  # sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
18
 
19
+ This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on the cppe-5 dataset.
20
  It achieves the following results on the evaluation set:
21
  - Loss: 1.0442
22
  - Map: 0.3514
 
61
  The following hyperparameters were used during training:
62
  - learning_rate: 5e-05
63
  - train_batch_size: 4
64
+ - eval_batch_size: 1
65
  - seed: 1337
66
  - gradient_accumulation_steps: 2
67
  - total_train_batch_size: 8
all_results.json CHANGED
@@ -26,35 +26,35 @@
26
  "eval_runtime": 15.6632,
27
  "eval_samples_per_second": 9.577,
28
  "eval_steps_per_second": 1.213,
29
- "test_loss": 0.7814600467681885,
30
- "test_map": 0.5279,
31
- "test_map_50": 0.8341,
32
- "test_map_75": 0.5627,
33
- "test_map_Coverall": 0.6264,
34
- "test_map_Face_Shield": 0.612,
35
- "test_map_Gloves": 0.4778,
36
- "test_map_Goggles": 0.4403,
37
- "test_map_Mask": 0.4831,
38
- "test_map_large": 0.705,
39
- "test_map_medium": 0.4585,
40
- "test_map_small": 0.5477,
41
- "test_mar_1": 0.3727,
42
- "test_mar_10": 0.621,
43
- "test_mar_100": 0.6349,
44
- "test_mar_100_Coverall": 0.7467,
45
- "test_mar_100_Face_Shield": 0.7471,
46
- "test_mar_100_Gloves": 0.5738,
47
- "test_mar_100_Goggles": 0.5281,
48
- "test_mar_100_Mask": 0.5788,
49
- "test_mar_large": 0.8137,
50
- "test_mar_medium": 0.5441,
51
- "test_mar_small": 0.5739,
52
- "test_runtime": 3.788,
53
- "test_samples_per_second": 7.656,
54
- "test_steps_per_second": 1.056,
55
  "total_flos": 2.1870706694064443e+19,
56
- "train_loss": 0.7253786786997093,
57
- "train_runtime": 17217.4822,
58
- "train_samples_per_second": 4.937,
59
- "train_steps_per_second": 0.616
60
  }
 
26
  "eval_runtime": 15.6632,
27
  "eval_samples_per_second": 9.577,
28
  "eval_steps_per_second": 1.213,
29
+ "test_loss": 1.2813646793365479,
30
+ "test_map": 0.3392,
31
+ "test_map_50": 0.8178,
32
+ "test_map_75": 0.1342,
33
+ "test_map_Coverall": 0.4236,
34
+ "test_map_Face_Shield": 0.366,
35
+ "test_map_Gloves": 0.3146,
36
+ "test_map_Goggles": 0.2668,
37
+ "test_map_Mask": 0.325,
38
+ "test_map_large": 0.4557,
39
+ "test_map_medium": 0.3312,
40
+ "test_map_small": 0.3208,
41
+ "test_mar_1": 0.2457,
42
+ "test_mar_10": 0.4543,
43
+ "test_mar_100": 0.4723,
44
+ "test_mar_100_Coverall": 0.5733,
45
+ "test_mar_100_Face_Shield": 0.5176,
46
+ "test_mar_100_Gloves": 0.4148,
47
+ "test_mar_100_Goggles": 0.425,
48
+ "test_mar_100_Mask": 0.4308,
49
+ "test_mar_large": 0.5709,
50
+ "test_mar_medium": 0.4222,
51
+ "test_mar_small": 0.3702,
52
+ "test_runtime": 5.3874,
53
+ "test_samples_per_second": 5.383,
54
+ "test_steps_per_second": 5.383,
55
  "total_flos": 2.1870706694064443e+19,
56
+ "train_loss": 0.0,
57
+ "train_runtime": 2.9311,
58
+ "train_samples_per_second": 28998.867,
59
+ "train_steps_per_second": 3616.329
60
  }
test_results.json CHANGED
@@ -1,29 +1,29 @@
1
  {
2
  "epoch": 99.53051643192488,
3
- "test_loss": 0.7814600467681885,
4
- "test_map": 0.5279,
5
- "test_map_50": 0.8341,
6
- "test_map_75": 0.5627,
7
- "test_map_Coverall": 0.6264,
8
- "test_map_Face_Shield": 0.612,
9
- "test_map_Gloves": 0.4778,
10
- "test_map_Goggles": 0.4403,
11
- "test_map_Mask": 0.4831,
12
- "test_map_large": 0.705,
13
- "test_map_medium": 0.4585,
14
- "test_map_small": 0.5477,
15
- "test_mar_1": 0.3727,
16
- "test_mar_10": 0.621,
17
- "test_mar_100": 0.6349,
18
- "test_mar_100_Coverall": 0.7467,
19
- "test_mar_100_Face_Shield": 0.7471,
20
- "test_mar_100_Gloves": 0.5738,
21
- "test_mar_100_Goggles": 0.5281,
22
- "test_mar_100_Mask": 0.5788,
23
- "test_mar_large": 0.8137,
24
- "test_mar_medium": 0.5441,
25
- "test_mar_small": 0.5739,
26
- "test_runtime": 3.788,
27
- "test_samples_per_second": 7.656,
28
- "test_steps_per_second": 1.056
29
  }
 
1
  {
2
  "epoch": 99.53051643192488,
3
+ "test_loss": 1.2813646793365479,
4
+ "test_map": 0.3392,
5
+ "test_map_50": 0.8178,
6
+ "test_map_75": 0.1342,
7
+ "test_map_Coverall": 0.4236,
8
+ "test_map_Face_Shield": 0.366,
9
+ "test_map_Gloves": 0.3146,
10
+ "test_map_Goggles": 0.2668,
11
+ "test_map_Mask": 0.325,
12
+ "test_map_large": 0.4557,
13
+ "test_map_medium": 0.3312,
14
+ "test_map_small": 0.3208,
15
+ "test_mar_1": 0.2457,
16
+ "test_mar_10": 0.4543,
17
+ "test_mar_100": 0.4723,
18
+ "test_mar_100_Coverall": 0.5733,
19
+ "test_mar_100_Face_Shield": 0.5176,
20
+ "test_mar_100_Gloves": 0.4148,
21
+ "test_mar_100_Goggles": 0.425,
22
+ "test_mar_100_Mask": 0.4308,
23
+ "test_mar_large": 0.5709,
24
+ "test_mar_medium": 0.4222,
25
+ "test_mar_small": 0.3702,
26
+ "test_runtime": 5.3874,
27
+ "test_samples_per_second": 5.383,
28
+ "test_steps_per_second": 5.383
29
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 99.53051643192488,
3
  "total_flos": 2.1870706694064443e+19,
4
- "train_loss": 0.7253786786997093,
5
- "train_runtime": 17217.4822,
6
- "train_samples_per_second": 4.937,
7
- "train_steps_per_second": 0.616
8
  }
 
1
  {
2
  "epoch": 99.53051643192488,
3
  "total_flos": 2.1870706694064443e+19,
4
+ "train_loss": 0.0,
5
+ "train_runtime": 2.9311,
6
+ "train_samples_per_second": 28998.867,
7
+ "train_steps_per_second": 3616.329
8
  }
trainer_state.json CHANGED
@@ -3712,10 +3712,10 @@
3712
  "epoch": 99.53051643192488,
3713
  "step": 10600,
3714
  "total_flos": 2.1870706694064443e+19,
3715
- "train_loss": 0.7253786786997093,
3716
- "train_runtime": 17217.4822,
3717
- "train_samples_per_second": 4.937,
3718
- "train_steps_per_second": 0.616
3719
  }
3720
  ],
3721
  "logging_steps": 500,
 
3712
  "epoch": 99.53051643192488,
3713
  "step": 10600,
3714
  "total_flos": 2.1870706694064443e+19,
3715
+ "train_loss": 0.0,
3716
+ "train_runtime": 2.9311,
3717
+ "train_samples_per_second": 28998.867,
3718
+ "train_steps_per_second": 3616.329
3719
  }
3720
  ],
3721
  "logging_steps": 500,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:44c466f6a4592c52de0708723d317a7bf185d350d17344bb5580710eb7bb0e8d
3
  size 4923
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:757bcf200175dacdb9978802ac15e01e66e778a24cd1ea2ff6d5573b7059cbdd
3
  size 4923