End of training

README.md

@@ -16,7 +16,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: 3.1388
+- Loss: 1.8007
 
 ## Model description
 
@@ -36,12 +36,12 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 4
-- eval_batch_size: 4
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- training_steps: 100
+- num_epochs: 25
 
 ### Training results
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:66178c66794116a565cee3cc60414c16c3882629c4168faf133170129887a9fb
+oid sha256:dce949a257a39bcc81c5f4674209c31159f67c2876d75445e530e2a8e82a133d
 size 166506812

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ba452b9dbbff0209fad7a3a349d32ca12733d707ad014636a4a1b5d8d83a37c
+oid sha256:f76f9ab901e24198f3553ef7b15b68c0113d37e67b7dcc51b2a24740cb1a82f2
 size 5240