End of training

README.md

@@ -3,8 +3,6 @@ license: apache-2.0
 base_model: t5-small
 tags:
 - generated_from_trainer
-metrics:
-- bleu
 model-index:
 - name: my_awesome_opus_books_model
   results: []
@@ -16,10 +14,6 @@ should probably proofread and complete it, then remove this comment. -->
 # my_awesome_opus_books_model
 
 This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 1.9866
-- Bleu: 57.2955
-- Gen Len: 3.1405
 
 ## Model description
 
@@ -40,12 +34,12 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
 - train_batch_size: 16
-- eval_batch_size: 16
+- eval_batch_size: 8
 - seed: 42
 - distributed_type: multi-GPU
-- num_devices: 2
-- total_train_batch_size: 32
-- total_eval_batch_size: 32
+- num_devices: 3
+- total_train_batch_size: 48
+- total_eval_batch_size: 24
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 2
@@ -53,10 +47,6 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
-| No log        | 1.0   | 348  | 2.1447          | 50.8861 | 3.1272  |
-| 2.8232        | 2.0   | 696  | 1.9866          | 57.2955 | 3.1405  |
 
 
 ### Framework versions

generation_config.json

@@ -1,4 +1,5 @@
 {
+  "_from_model_config": true,
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "pad_token_id": 0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a14d270b9e164803aef6627a2a305b202d3bd11744d7529ddeca440a7037c7e0
+oid sha256:12840949a4668b92b91777b0a9ebc150a3d92f313795838d4bbc3a20f60d238f
 size 242041896