rcook commited on
Commit
30c379f
·
verified ·
1 Parent(s): a4d8d22

End of training

Browse files
README.md CHANGED
@@ -1,9 +1,11 @@
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
- base_model: google-t5/t5-small
5
  tags:
6
  - generated_from_trainer
 
 
7
  model-index:
8
  - name: tmp_test
9
  results: []
@@ -14,7 +16,14 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # tmp_test
16
 
17
- This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
 
 
 
 
 
 
 
18
 
19
  ## Model description
20
 
@@ -33,20 +42,36 @@ More information needed
33
  ### Training hyperparameters
34
 
35
  The following hyperparameters were used during training:
36
- - learning_rate: 2e-05
37
  - train_batch_size: 16
38
  - eval_batch_size: 16
39
  - seed: 42
 
 
40
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
41
  - lr_scheduler_type: linear
42
- - num_epochs: 2
 
43
 
44
  ### Training results
45
 
 
 
 
 
 
 
 
 
 
 
 
 
46
 
47
 
48
  ### Framework versions
49
 
50
  - Transformers 4.48.3
51
  - Pytorch 2.5.1+cu124
 
52
  - Tokenizers 0.21.0
 
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
+ base_model: t5-small
5
  tags:
6
  - generated_from_trainer
7
+ metrics:
8
+ - rouge
9
  model-index:
10
  - name: tmp_test
11
  results: []
 
16
 
17
  # tmp_test
18
 
19
+ This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 3.4968
22
+ - Rouge1: 0.1438
23
+ - Rouge2: 0.0337
24
+ - Rougel: 0.1049
25
+ - Rougelsum: 0.1052
26
+ - Gen Len: 20.0
27
 
28
  ## Model description
29
 
 
42
  ### Training hyperparameters
43
 
44
  The following hyperparameters were used during training:
45
+ - learning_rate: 1e-05
46
  - train_batch_size: 16
47
  - eval_batch_size: 16
48
  - seed: 42
49
+ - gradient_accumulation_steps: 2
50
+ - total_train_batch_size: 32
51
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
  - lr_scheduler_type: linear
53
+ - num_epochs: 10
54
+ - mixed_precision_training: Native AMP
55
 
56
  ### Training results
57
 
58
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
59
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
60
+ | No log | 1.0 | 25 | 4.3210 | 0.1303 | 0.0263 | 0.0973 | 0.0974 | 20.0 |
61
+ | No log | 2.0 | 50 | 3.8167 | 0.1344 | 0.0288 | 0.1005 | 0.1005 | 20.0 |
62
+ | No log | 3.0 | 75 | 3.6506 | 0.1386 | 0.0305 | 0.1023 | 0.1024 | 20.0 |
63
+ | No log | 4.0 | 100 | 3.5756 | 0.1426 | 0.0317 | 0.1044 | 0.1045 | 20.0 |
64
+ | No log | 5.0 | 125 | 3.5415 | 0.1433 | 0.0323 | 0.1044 | 0.1045 | 20.0 |
65
+ | No log | 6.0 | 150 | 3.5215 | 0.1425 | 0.0321 | 0.1039 | 0.104 | 20.0 |
66
+ | No log | 7.0 | 175 | 3.5101 | 0.1436 | 0.0325 | 0.1047 | 0.1048 | 20.0 |
67
+ | 3.9618 | 8.0 | 200 | 3.5025 | 0.1435 | 0.0331 | 0.105 | 0.1051 | 20.0 |
68
+ | 3.9618 | 9.0 | 225 | 3.4982 | 0.1438 | 0.0337 | 0.1049 | 0.1052 | 20.0 |
69
+ | 3.9618 | 10.0 | 250 | 3.4968 | 0.1438 | 0.0337 | 0.1049 | 0.1052 | 20.0 |
70
 
71
 
72
  ### Framework versions
73
 
74
  - Transformers 4.48.3
75
  - Pytorch 2.5.1+cu124
76
+ - Datasets 3.3.2
77
  - Tokenizers 0.21.0
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google-t5/t5-small",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
 
1
  {
2
+ "_name_or_path": "t5-small",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
generation_config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_from_model_config": true,
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
 
1
  {
 
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:06ceeeae367f40dd81599ed3e2c5baa832f8f8f78ec879641dc2d0429a3e6a85
3
  size 242041896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53d76fde505168d38d623c71361304830a91672cb31c9dd86ee6a55e341e126d
3
  size 242041896
runs/Feb24_20-12-06_94086e04c5b2/events.out.tfevents.1740427929.94086e04c5b2.318.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4ed9abb0378189bd5ad6df491cd7b2a0924dadb876d2e4c88bbbc3059c5e334
3
+ size 7908
runs/Feb24_20-12-06_94086e04c5b2/events.out.tfevents.1740428950.94086e04c5b2.318.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5c1636e26f83e713c183d5785aa5f40732c0be9b2ad29d645e1095528951691
3
+ size 603
runs/Feb24_20-57-52_94086e04c5b2/events.out.tfevents.1740430677.94086e04c5b2.32775.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7016430c106f3401978da7fb2adeb84a21218d3268d12d92ab4e0d4515e6dd0b
3
+ size 11783
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c29e89311a67645a7b21bd8b131acc08eabbf989f2e8ac3f3afa2b34d911c463
3
- size 5432
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd6474827b93beaa728259a765b19d4b5448d7a9ec2a1099e78cd17db25fa03b
3
+ size 5496