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 [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2350
+- Loss: 0.1869
 
 ## Model description
 
@@ -36,21 +36,23 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 8
+- train_batch_size: 16
 - eval_batch_size: 16
 - 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
 - lr_scheduler_warmup_steps: 500
-- num_epochs: 3
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.4461        | 1.0   | 2798 | 0.2324          |
-| 0.1122        | 2.0   | 5596 | 0.2321          |
-| 0.1304        | 3.0   | 8394 | 0.2350          |
+| 0.22          | 1.0   | 1399 | 0.1804          |
+| 0.1415        | 2.0   | 2798 | 0.1627          |
+| 0.1872        | 3.0   | 4197 | 0.1736          |
+| 0.1603        | 4.0   | 5596 | 0.1807          |
+| 0.1039        | 5.0   | 6995 | 0.1869          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c98bfda0397e56669d7d07936b8ef4ae29eddb41552f4f0a605b295358d04ab9
+oid sha256:2ac122e185891239479505c09a438d62dc6ab920137c05fba209133858b55960
 size 17549312

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:41e93b8246e33edb3aacd15037b594f876feb32f633c5c9a4614ef3eae667733
+oid sha256:9f18dbb661b33ef53aa9729ac3c90da3e0b811a83f2d54085432f4a4a3f5d07f
 size 5368