End of training

README.md

@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) on the eli5_category dataset.
 It achieves the following results on the evaluation set:
-- Loss: 3.5890
+- Loss: 3.6474
 
 ## Model description
 
@@ -38,25 +38,25 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 32
-- eval_batch_size: 32
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 3.0
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 3.5937        | 1.0   | 2000 | 3.5904          |
-| 3.4515        | 2.0   | 4000 | 3.5851          |
-| 3.3591        | 3.0   | 6000 | 3.5890          |
+| 3.6611        | 1.0   | 1301 | 3.6505          |
+| 3.5104        | 2.0   | 2602 | 3.6461          |
+| 3.4318        | 3.0   | 3903 | 3.6474          |
 
 
 ### Framework versions
 
-- Transformers 4.46.2
+- Transformers 4.46.3
 - Pytorch 2.5.1+cu121
 - Datasets 3.1.0
 - Tokenizers 0.20.3

generation_config.json

@@ -2,5 +2,5 @@
   "_from_model_config": true,
   "bos_token_id": 50256,
   "eos_token_id": 50256,
-  "transformers_version": "4.46.2"
+  "transformers_version": "4.46.3"
 }

runs/Dec10_12-05-42_31f25b1feafd/events.out.tfevents.1733832354.31f25b1feafd.3818.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c9850c359068b1a76cbfb259119addf62f1d37a370e48dec64c3e1936586cc63
-size 7535
+oid sha256:25fd732ede9856f2d561c492cc1c020eb48f07af08e4df4c743aa7a38b2ba6f3
+size 8160

runs/Dec10_12-05-42_31f25b1feafd/events.out.tfevents.1733833181.31f25b1feafd.3818.1

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