End of training

README.md

@@ -13,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # detr-resnet-50_finetuned_detect-waste
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.7927
+- Loss: 1.5678
 - Dummy: 1
 
 ## Model description
@@ -35,24 +35,35 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
-- train_batch_size: 8
-- eval_batch_size: 8
+- learning_rate: 1e-05
+- train_batch_size: 4
+- eval_batch_size: 4
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: polynomial
-- num_epochs: 2
+- lr_scheduler_type: linear
+- num_epochs: 100
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Dummy |
 |:-------------:|:-----:|:----:|:---------------:|:-----:|
-| 2.081         | 1.95  | 800  | 1.7927          | 1     |
+| 2.4218        | 5.26  | 500  | 2.0445          | 1     |
+| 2.269         | 10.53 | 1000 | 1.9227          | 1     |
+| 1.9898        | 15.79 | 1500 | 1.7609          | 1     |
+| 2.0422        | 21.05 | 2000 | 1.6763          | 1     |
+| 2.1387        | 26.32 | 2500 | 1.7746          | 1     |
+| 1.8163        | 31.58 | 3000 | 1.6525          | 1     |
+| 1.7993        | 36.84 | 3500 | 1.6010          | 1     |
+| 1.8009        | 42.11 | 4000 | 1.5959          | 1     |
+| 1.8148        | 47.37 | 4500 | 1.5332          | 1     |
+| 1.7837        | 52.63 | 5000 | 1.5525          | 1     |
+| 1.5934        | 57.89 | 5500 | 1.5409          | 1     |
+| 1.5357        | 63.16 | 6000 | 1.5678          | 1     |
 
 
 ### Framework versions
 
-- Transformers 4.32.0
+- Transformers 4.33.0.dev0
 - Pytorch 2.0.1+cu117
 - Datasets 2.14.4
 - Tokenizers 0.13.3

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:07c502db571359bd6fbe895dcc58d30310d4a547cf34839a097506ed0500eedd
+oid sha256:606cc3ec84df58a6ac8d1913d375d4398168a286c06fe400d5ab04ec0ab8b853
 size 166620821