End of training

README.md

@@ -19,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2461
-- Accuracy: 0.9197
-- F1: 0.9197
+- Loss: 0.3198
+- Accuracy: 0.9211
+- F1: 0.9209
 
 ## Model description
 
@@ -41,8 +41,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 32
-- eval_batch_size: 32
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
@@ -52,8 +52,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| No log        | 1.0   | 313  | 0.2724          | 0.9093   | 0.9091 |
-| 0.3091        | 2.0   | 626  | 0.2461          | 0.9197   | 0.9197 |
+| 0.3584        | 1.0   | 1250 | 0.3159          | 0.9095   | 0.9094 |
+| 0.1976        | 2.0   | 2500 | 0.3198          | 0.9211   | 0.9209 |
 
 
 ### Framework versions

runs/Sep22_21-03-08_121c5192df5b/events.out.tfevents.1758574995.121c5192df5b.577.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5650e087a74df0e59b879db7f8c9c0221574d4ef4c1940107df7e23ca62846bc
-size 6482
+oid sha256:7edc6feaf98cc2740dacce5e62c790b36d8169259e352c0bcbb4cc25af00112d
+size 7205