Transformers
TensorBoard
Safetensors
t5
text2text-generation
Generated from Trainer
text-generation-inference
Instructions to use dany0407/summarization_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dany0407/summarization_model with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("dany0407/summarization_model") model = AutoModelForSeq2SeqLM.from_pretrained("dany0407/summarization_model", device_map="auto") - Notebooks
- Google Colab
- Kaggle
End of training
Browse files
README.md
CHANGED
|
@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
-
- Loss: 2.
|
| 22 |
-
- Rouge1: 0.
|
| 23 |
-
- Rouge2: 0.
|
| 24 |
-
- Rougel: 0.
|
| 25 |
-
- Rougelsum: 0.
|
| 26 |
- Gen Len: 20.0
|
| 27 |
|
| 28 |
## Model description
|
|
@@ -55,15 +55,15 @@ The following hyperparameters were used during training:
|
|
| 55 |
|
| 56 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|
| 57 |
|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
|
| 58 |
-
| No log | 1.0 | 62 | 2.
|
| 59 |
-
| No log | 2.0 | 124 | 2.
|
| 60 |
-
| No log | 3.0 | 186 | 2.
|
| 61 |
-
| No log | 4.0 | 248 | 2.
|
| 62 |
|
| 63 |
|
| 64 |
### Framework versions
|
| 65 |
|
| 66 |
-
- Transformers 4.
|
| 67 |
-
- Pytorch 2.
|
| 68 |
- Datasets 4.0.0
|
| 69 |
-
- Tokenizers 0.21.
|
|
|
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 2.5574
|
| 22 |
+
- Rouge1: 0.1496
|
| 23 |
+
- Rouge2: 0.0554
|
| 24 |
+
- Rougel: 0.1221
|
| 25 |
+
- Rougelsum: 0.1221
|
| 26 |
- Gen Len: 20.0
|
| 27 |
|
| 28 |
## Model description
|
|
|
|
| 55 |
|
| 56 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|
| 57 |
|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
|
| 58 |
+
| No log | 1.0 | 62 | 2.8486 | 0.1367 | 0.041 | 0.1133 | 0.1132 | 20.0 |
|
| 59 |
+
| No log | 2.0 | 124 | 2.6347 | 0.148 | 0.0525 | 0.1198 | 0.12 | 20.0 |
|
| 60 |
+
| No log | 3.0 | 186 | 2.5750 | 0.1484 | 0.0551 | 0.1208 | 0.121 | 20.0 |
|
| 61 |
+
| No log | 4.0 | 248 | 2.5574 | 0.1496 | 0.0554 | 0.1221 | 0.1221 | 20.0 |
|
| 62 |
|
| 63 |
|
| 64 |
### Framework versions
|
| 65 |
|
| 66 |
+
- Transformers 4.53.2
|
| 67 |
+
- Pytorch 2.10.0+cu128
|
| 68 |
- Datasets 4.0.0
|
| 69 |
+
- Tokenizers 0.21.4
|
generation_config.json
CHANGED
|
@@ -2,5 +2,5 @@
|
|
| 2 |
"decoder_start_token_id": 0,
|
| 3 |
"eos_token_id": 1,
|
| 4 |
"pad_token_id": 0,
|
| 5 |
-
"transformers_version": "4.
|
| 6 |
}
|
|
|
|
| 2 |
"decoder_start_token_id": 0,
|
| 3 |
"eos_token_id": 1,
|
| 4 |
"pad_token_id": 0,
|
| 5 |
+
"transformers_version": "4.53.2"
|
| 6 |
}
|
runs/May13_04-17-40_c586c8e97feb/events.out.tfevents.1778645864.c586c8e97feb.3848.0
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a0519e1c2c2938baacb58fb74dc6e72e16dcbd33e4ce6d2a947843c2000b8d06
|
| 3 |
+
size 8406
|