Hugging Face's logo

chamdentimem
/
ViT5_Dialogue_Summarization

Text2Text Generation
Transformers
TensorBoard
Safetensors
t5
Generated from Trainer
text-generation-inference
Inference Endpoints
Model card Files Metrics Community
chamdentimem commited on
Commit
27bc05b
•
1 Parent(s): f3cb39c

End of training

Browse files
Files changed (8) hide show
  1. README.md +13 -15
  2. logs/events.out.tfevents.1711099495.5da5938990cd.1513.0 +3 -0
  3. logs/events.out.tfevents.1711099590.5da5938990cd.4571.0 +3 -0
  4. logs/events.out.tfevents.1711099755.5da5938990cd.5456.0 +3 -0
  5. logs/events.out.tfevents.1711105019.5da5938990cd.5456.1 +3 -0
  6. model.safetensors +1 -1
  7. tokenizer.json +1 -6
  8. training_args.bin +1 -1
README.md CHANGED
@@ -17,12 +17,12 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [VietAI/vit5-base](https://huggingface.co/VietAI/vit5-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.6228
21
- - Rouge1: 52.8317
22
- - Rouge2: 29.8327
23
- - Rougel: 41.7287
24
- - Rougelsum: 46.3648
25
- - Gen Len: 16.8071
26
 
27
  ## Model description
28
 
@@ -42,25 +42,23 @@ More information needed
42
 
43
  The following hyperparameters were used during training:
44
  - learning_rate: 5e-05
45
- - train_batch_size: 4
46
- - eval_batch_size: 4
47
  - seed: 42
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
- - num_epochs: 3
51
 
52
  ### Training results
53
 
54
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
- |:-------------:|:-----:|:-----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
56
- | 1.8522 | 1.0 | 3683 | 1.6994 | 52.1383 | 28.4438 | 41.2886 | 45.7379 | 16.6484 |
57
- | 1.496 | 2.0 | 7366 | 1.6228 | 52.8317 | 29.8327 | 41.7287 | 46.3648 | 16.8071 |
58
- | 1.1959 | 3.0 | 11049 | 1.6419 | 53.6738 | 30.2857 | 42.5458 | 47.2315 | 16.906 |
59
 
60
 
61
  ### Framework versions
62
 
63
  - Transformers 4.38.2
64
- - Pytorch 2.1.0+cu121
65
  - Datasets 2.18.0
66
  - Tokenizers 0.15.2
 
17
 
18
  This model is a fine-tuned version of [VietAI/vit5-base](https://huggingface.co/VietAI/vit5-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.6377
21
+ - Rouge1: 51.4583
22
+ - Rouge2: 28.5387
23
+ - Rougel: 40.9957
24
+ - Rougelsum: 45.4297
25
+ - Gen Len: 16.1661
26
 
27
  ## Model description
28
 
 
42
 
43
  The following hyperparameters were used during training:
44
  - learning_rate: 5e-05
45
+ - train_batch_size: 2
46
+ - eval_batch_size: 2
47
  - seed: 42
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
+ - num_epochs: 1
51
 
52
  ### Training results
53
 
54
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
56
+ | 0.5898 | 1.0 | 7366 | 0.6377 | 51.4583 | 28.5387 | 40.9957 | 45.4297 | 16.1661 |
 
 
57
 
58
 
59
  ### Framework versions
60
 
61
  - Transformers 4.38.2
62
+ - Pytorch 2.2.1+cu121
63
  - Datasets 2.18.0
64
  - Tokenizers 0.15.2
logs/events.out.tfevents.1711099495.5da5938990cd.1513.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c0be40f3c6a5bfe5f4790c186fb7dc3a2e80fd04f7b3cf3744370f3c8b76363
3
+ size 4184
logs/events.out.tfevents.1711099590.5da5938990cd.4571.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b152041cccffac41f5f6329381e479af38526741513d333825b0bb8808ad5895
3
+ size 4184
logs/events.out.tfevents.1711099755.5da5938990cd.5456.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1df0e736edbf9b5950379987298065e0d95c9069402f6ee45a10615bee3436dc
3
+ size 8704
logs/events.out.tfevents.1711105019.5da5938990cd.5456.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7498c4ab4208bf549a937a4ea13bbb96beef3c66dcc1f6617aefbc418bebd05
3
+ size 613
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:97c2f41aa1cd18159dce5ec670d14acf413d802085d0150497ff89c5a2a7d155
3
  size 903834408
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e857362389e0800797a39f2b3a0184284fcf38cf96d34b3822c132d6ab3f01de
3
  size 903834408
tokenizer.json CHANGED
@@ -1,11 +1,6 @@
1
  {
2
  "version": "1.0",
3
- "truncation": {
4
- "direction": "Right",
5
- "max_length": 256,
6
- "strategy": "LongestFirst",
7
- "stride": 0
8
- },
9
  "padding": null,
10
  "added_tokens": [
11
  {
 
1
  {
2
  "version": "1.0",
3
+ "truncation": null,
 
 
 
 
 
4
  "padding": null,
5
  "added_tokens": [
6
  {
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1acfa877f70695b67842fb330f4e2441899e2ad874a9538a1e8592ff4975bbd
3
  size 5048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06f5b44ebab1040d112433baf372bb8136b4207a80d14ff45748180625112e55
3
  size 5048