End of training

README.md

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.3509
+- Loss: 1.7604
 
 ## Model description
 
@@ -41,15 +41,18 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 100
-- num_epochs: 1
+- num_epochs: 2
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 3.7305        | 0.4   | 500  | 2.8254          |
-| 2.403         | 0.8   | 1000 | 2.3509          |
+| 3.6061        | 0.4   | 500  | 2.5979          |
+| 2.2323        | 0.8   | 1000 | 2.0724          |
+| 1.9294        | 1.2   | 1500 | 1.8745          |
+| 1.8162        | 1.6   | 2000 | 1.7907          |
+| 1.7529        | 2.0   | 2500 | 1.7604          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6b9471bd3b24474bcc42e1fe12da03a1794d84762d7045482c0316468dde702c
+oid sha256:37dc7b2e3d8ede29a88ae044eabbab866c81b94827069e59deffcb234f3416c0
 size 166506812

runs/Apr21_13-47-13_7ef39fe90403/events.out.tfevents.1713707234.7ef39fe90403.168.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:353ab1d53bc157c929af6d691ed1f27c36600f9ff25e2d5f18a0ab3f0112408e
-size 9302
+oid sha256:d3538523bab7484cb2a56ba02de51a50df38dc8a74490d66b3b682e0313ccafe
+size 10138