End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: philschmid/flan-t5-base-samsum
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: flan-t5-base-samsum-spotify-podcasts
   results: []
@@ -14,6 +16,13 @@ should probably proofread and complete it, then remove this comment. -->
 # flan-t5-base-samsum-spotify-podcasts
 
 This model is a fine-tuned version of [philschmid/flan-t5-base-samsum](https://huggingface.co/philschmid/flan-t5-base-samsum) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.3026
+- Rouge1: 0.27
+- Rouge2: 0.1512
+- Rougel: 0.2352
+- Rougelsum: 0.2355
+- Gen Len: 19.0
 
 ## Model description
 
@@ -33,24 +42,27 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- train_batch_size: 4
+- eval_batch_size: 4
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 1
+- num_epochs: 4
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
-| No log        | 1.0   | 466  | nan             | 0.1911 | 0.0819 | 0.1582 | 0.158     | 18.5279 |
+| No log        | 1.0   | 233  | 1.3963          | 0.2419 | 0.1266 | 0.2079 | 0.2077    | 19.0    |
+| No log        | 2.0   | 466  | 1.3356          | 0.2637 | 0.1432 | 0.2265 | 0.2263    | 19.0    |
+| 1.6496        | 3.0   | 699  | 1.3088          | 0.2695 | 0.1491 | 0.2331 | 0.2331    | 19.0    |
+| 1.6496        | 4.0   | 932  | 1.3026          | 0.27   | 0.1512 | 0.2352 | 0.2355    | 19.0    |
 
 
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 2.1.0+cu118
-- Datasets 2.15.0
-- Tokenizers 0.15.0
+- Transformers 4.35.0
+- Pytorch 2.0.0
+- Datasets 2.1.0
+- Tokenizers 0.14.1

generation_config.json

@@ -2,5 +2,5 @@
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "pad_token_id": 0,
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.35.0"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8aea98097bbe7257f45ffc50d4e10a7fea32b5fd21bc78315f7b84d556705176
+oid sha256:681ff07382d71003671b017b561a5c5923b693b5c66759130b34d8adaebe7303
 size 990345064

runs/Nov18_17-06-56_6c367cafd2b8/events.out.tfevents.1700327221.6c367cafd2b8.47.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:565f49aeb6e3267553b1339e497c4dac921a5233bbd6124d051b321c21b84d00
-size 6564
+oid sha256:329574d9dc1d0735ecf5315e446df38da3c2b2de10c697d9cbbc248f59c1ab72
+size 7968