Model save

README.md

@@ -15,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0924
-- Rouge1: 34.7587
-- Rouge2: 25.3644
-- Rougel: 34.5531
-- Rougelsum: 34.6322
-- Gen Len: 13.584
+- Loss: 0.2334
+- Rouge1: 27.4942
+- Rouge2: 21.9051
+- Rougel: 26.9521
+- Rougelsum: 27.0238
+- Gen Len: 16.0
 
 ## Model description
 
@@ -40,8 +40,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 4
-- eval_batch_size: 4
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -51,9 +51,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
-| 0.0952        | 1.0   | 2500 | 0.0960          | 26.9673 | 20.9727 | 26.939  | 26.9315   | 15.552  |
-| 0.0768        | 2.0   | 5000 | 0.0931          | 35.4227 | 25.9227 | 35.1707 | 35.3002   | 13.552  |
-| 0.0662        | 3.0   | 7500 | 0.0924          | 34.7587 | 25.3644 | 34.5531 | 34.6322   | 13.584  |
+| No log        | 1.0   | 125  | 0.3533          | 0.0     | 0.0     | 0.0     | 0.0       | 20.0    |
+| No log        | 2.0   | 250  | 0.3009          | 7.9737  | 1.5     | 7.7593  | 7.9222    | 16.38   |
+| No log        | 3.0   | 375  | 0.2334          | 27.4942 | 21.9051 | 26.9521 | 27.0238   | 16.0    |
 
 
 ### Framework versions