End of training

README.md

@@ -32,16 +32,16 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.00047
-- train_batch_size: 163
-- eval_batch_size: 163
+- learning_rate: 0.0005
+- train_batch_size: 32
+- eval_batch_size: 32
 - seed: 42
 - gradient_accumulation_steps: 8
-- total_train_batch_size: 1304
+- total_train_batch_size: 256
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_steps: 1000
-- num_epochs: 2
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
@@ -50,7 +50,7 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 2.1.0+cu121
-- Datasets 2.16.1
-- Tokenizers 0.15.0
+- Transformers 4.44.0
+- Pytorch 2.4.0+cu121
+- Datasets 3.5.0
+- Tokenizers 0.19.1

generation_config.json

@@ -2,5 +2,5 @@
   "_from_model_config": true,
   "bos_token_id": 50256,
   "eos_token_id": 50256,
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.44.0"
 }