Hugging Face's logo

rizvi-rahil786
/
t5-small-shShootingKP

Text2Text Generation
Transformers
TensorBoard
Safetensors
t5
Generated from Trainer
text-generation-inference
Inference Endpoints
Model card Files Metrics Community
rizvi-rahil786 commited on
Commit
e30dc35
•
1 Parent(s): 8a6185a

End of training

Browse files
Files changed (7) hide show
  1. README.md +18 -15
  2. config.json +1 -1
  3. generation_config.json +1 -1
  4. logs/events.out.tfevents.1712422001.c95cc94d97f3.2364.0 +3 -0
  5. logs/events.out.tfevents.1712431267.c95cc94d97f3.2364.1 +3 -0
  6. model.safetensors +1 -1
  7. training_args.bin +2 -2
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 [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.6515
21
- - Rouge1: 33.8635
22
- - Rouge2: 27.4791
23
- - Rougel: 33.8476
24
- - Rougelsum: 33.8364
25
- - Gen Len: 6.7861
26
 
27
  ## Model description
28
 
@@ -42,27 +42,30 @@ More information needed
42
 
43
  The following hyperparameters were used during training:
44
  - learning_rate: 5e-05
45
- - train_batch_size: 8
46
- - eval_batch_size: 8
47
  - seed: 42
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
- - num_epochs: 5
51
 
52
  ### Training results
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
  |:-------------:|:-----:|:-----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
56
- | 1.2344 | 1.0 | 3028 | 1.7377 | 35.027 | 28.4749 | 35.0561 | 34.9832 | 6.975 |
57
- | 1.0665 | 2.0 | 6056 | 1.6887 | 33.9581 | 27.5342 | 33.9114 | 33.8885 | 6.5995 |
58
- | 1.0313 | 3.0 | 9084 | 1.6515 | 33.8635 | 27.4791 | 33.8476 | 33.8364 | 6.7861 |
59
- | 0.9957 | 4.0 | 12112 | 1.6793 | 34.3439 | 27.7857 | 34.3413 | 34.3188 | 6.7572 |
60
- | 0.9688 | 5.0 | 15140 | 1.6570 | 34.2704 | 27.6975 | 34.2984 | 34.2296 | 6.7659 |
 
 
 
61
 
62
 
63
  ### Framework versions
64
 
65
- - Transformers 4.39.1
66
  - Pytorch 2.2.1+cu121
67
  - Datasets 2.18.0
68
  - Tokenizers 0.15.2
 
17
 
18
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 1.6665
21
+ - Rouge1: 34.5395
22
+ - Rouge2: 28.1355
23
+ - Rougel: 34.511
24
+ - Rougelsum: 34.5702
25
+ - Gen Len: 6.4385
26
 
27
  ## Model description
28
 
 
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: 8
51
 
52
  ### Training results
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
  |:-------------:|:-----:|:-----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
56
+ | 1.1889 | 1.0 | 6056 | 1.7462 | 35.0128 | 28.2062 | 34.9183 | 34.9898 | 7.2062 |
57
+ | 1.0076 | 2.0 | 12112 | 1.7262 | 34.4492 | 27.6944 | 34.3991 | 34.4121 | 6.8452 |
58
+ | 0.9798 | 3.0 | 18168 | 1.6861 | 34.5767 | 28.1063 | 34.5199 | 34.5759 | 6.6332 |
59
+ | 0.9324 | 4.0 | 24224 | 1.7051 | 34.2272 | 27.9947 | 34.2424 | 34.2402 | 6.8740 |
60
+ | 0.8892 | 5.0 | 30280 | 1.6665 | 34.5395 | 28.1355 | 34.511 | 34.5702 | 6.4385 |
61
+ | 0.809 | 6.0 | 36336 | 1.7787 | 34.0894 | 27.6763 | 34.0998 | 34.1103 | 6.6721 |
62
+ | 0.7897 | 7.0 | 42392 | 1.7404 | 33.8737 | 27.6484 | 33.8941 | 33.9224 | 6.5274 |
63
+ | 0.7302 | 8.0 | 48448 | 1.7334 | 33.7942 | 27.6188 | 33.8028 | 33.8252 | 6.4447 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - Transformers 4.39.3
69
  - Pytorch 2.2.1+cu121
70
  - Datasets 2.18.0
71
  - Tokenizers 0.15.2
config.json CHANGED
@@ -55,7 +55,7 @@
55
  }
56
  },
57
  "torch_dtype": "float32",
58
- "transformers_version": "4.39.1",
59
  "use_cache": true,
60
  "vocab_size": 32128
61
  }
 
55
  }
56
  },
57
  "torch_dtype": "float32",
58
+ "transformers_version": "4.39.3",
59
  "use_cache": true,
60
  "vocab_size": 32128
61
  }
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.39.1"
6
  }
 
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
5
+ "transformers_version": "4.39.3"
6
  }
logs/events.out.tfevents.1712422001.c95cc94d97f3.2364.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:596bb6533363111af91443975e0809e9eb1cc87c4878a28f80f707583a6102e6
3
+ size 30715
logs/events.out.tfevents.1712431267.c95cc94d97f3.2364.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3038893d7a70724065152b15c5b39a1626b684f2e3490e5cd15aa1c81cb2ebda
3
+ size 623
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e661afc5e1691502bf8943cab23dc9e94886c970c4387a7f65c37b07534f952
3
  size 242041896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a431fc7b98a0baf5ce805a3d1abb0390f69498527ca59dd4cf2fee09751720e8
3
  size 242041896
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c90fa406f22e7f6a58ab5fcf237ec4e19f58d67fa43599ac762a31900cc1de81
3
- size 5048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0299ef4492e00b979c74ad2c539fbb74eb5fe027bb5ad312cfc15e38c84e12d
3
+ size 5112