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: 1.5074
+- Loss: 1.3326
 
 ## Model description
 
@@ -35,11 +35,9 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
-- train_batch_size: 16
+- train_batch_size: 32
 - eval_batch_size: 8
 - seed: 42
-- gradient_accumulation_steps: 2
-- total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 1
@@ -49,9 +47,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 2.5836        | 0.32  | 100  | 1.7905          |
-| 1.6993        | 0.64  | 200  | 1.5704          |
-| 1.571         | 0.96  | 300  | 1.5074          |
+| 2.5582        | 0.32  | 100  | 1.7392          |
+| 1.5953        | 0.64  | 200  | 1.4379          |
+| 1.4199        | 0.96  | 300  | 1.3326          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:df0e3f7e308f9c0924b4d2c395b7bc0db09d794f88a43a38e4f72cc1f17eed09
+oid sha256:98cd088bc159f64a38294d9799c2973c403746ecbeb0e6c4dc70e2e84bea1a33
 size 166506812

runs/Apr17_13-57-10_499bb3dcf1d7/events.out.tfevents.1713362262.499bb3dcf1d7.510.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1984f03b04d2a7053590fa8e7c82a1584e0f30da5930ac487e490ac40c7638ae
+size 9167

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:81ad9ae58676b2e050601385a3972674df8a694e9b9823ee77514ea67c3be548
+oid sha256:5e45513570aa83ba31b1004035fd9b4da6c838e40a88782993d7b6c83e12a996
 size 4856