gorgor1000 commited on
Commit
f7eefaf
·
verified ·
1 Parent(s): 5bcdffd

End of training

Browse files
Files changed (3) hide show
  1. README.md +13 -4
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 3.4259
20
 
21
  ## Model description
22
 
@@ -35,19 +35,28 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 1e-05
39
  - train_batch_size: 2
40
  - eval_batch_size: 2
41
  - seed: 42
42
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
  - lr_scheduler_type: linear
44
- - num_epochs: 1
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 3.2876 | 1.0 | 500 | 3.1846 |
 
 
 
 
 
 
 
 
 
51
 
52
 
53
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.2560
20
 
21
  ## Model description
22
 
 
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
+ - learning_rate: 0.0001
39
  - train_batch_size: 2
40
  - eval_batch_size: 2
41
  - seed: 42
42
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
  - lr_scheduler_type: linear
44
+ - num_epochs: 10
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 1.9388 | 1.0 | 500 | 1.6931 |
51
+ | 1.6687 | 2.0 | 1000 | 1.6105 |
52
+ | 1.6005 | 3.0 | 1500 | 1.4854 |
53
+ | 1.4927 | 4.0 | 2000 | 1.4709 |
54
+ | 1.433 | 5.0 | 2500 | 1.3741 |
55
+ | 1.3916 | 6.0 | 3000 | 1.3862 |
56
+ | 1.3196 | 7.0 | 3500 | 1.3261 |
57
+ | 1.2961 | 8.0 | 4000 | 1.2754 |
58
+ | 1.25 | 9.0 | 4500 | 1.2689 |
59
+ | 1.2318 | 10.0 | 5000 | 1.2476 |
60
 
61
 
62
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cb9072df1fc685945bede0129d17d1dc17d80ee25f3f73ecb399109524cec334
3
  size 166506812
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db3a46b515ac92efc7dcb9025a46f85294585f59107f568407ade41697cbf551
3
  size 166506812
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4d3e483cccbdf13dded7a14c6bfe27ed33615167092f480db0ef3b7b8f730404
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9273055c4fd092306b653fe4bb2686dc30b97b0f165e8430c1282ce3ea81a2e
3
  size 5304