owenMills commited on
Commit
5926a70
·
verified ·
1 Parent(s): 27d3bdc

End of training

Browse files
README.md CHANGED
@@ -1,16 +1,14 @@
1
- ---
2
- library_name: peft
3
- base_model: owenMills/bart-pas-model
4
- tags:
5
- - base_model:adapter:owenMills/bart-pas-model
6
- - lora
7
- - transformers
8
- metrics:
9
- - rouge
10
- model-index:
11
- - name: bart-pas-lora
12
- results: []
13
- ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
  should probably proofread and complete it, then remove this comment. -->
@@ -19,11 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [owenMills/bart-pas-model](https://huggingface.co/owenMills/bart-pas-model) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 2.6618
23
- - Rouge1: 46.92
24
- - Rouge2: 38.48
25
- - Rougel: 43.2
26
- - Rougelsum: 43.34
27
 
28
  ## Model description
29
 
@@ -42,29 +40,33 @@ More information needed
42
  ### Training hyperparameters
43
 
44
  The following hyperparameters were used during training:
45
- - learning_rate: 5e-05
46
  - train_batch_size: 2
47
  - eval_batch_size: 2
48
  - seed: 42
49
  - gradient_accumulation_steps: 8
50
  - total_train_batch_size: 16
51
- - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
  - lr_scheduler_type: linear
53
  - lr_scheduler_warmup_ratio: 0.05
54
- - num_epochs: 1
 
55
  - label_smoothing_factor: 0.1
56
 
57
  ### Training results
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
60
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
61
- | 4.4614 | 1.0 | 13 | 2.6618 | 46.92 | 38.48 | 43.2 | 43.34 |
 
 
 
62
 
63
 
64
  ### Framework versions
65
 
66
- - PEFT 0.17.1
67
- - Transformers 4.57.1
68
- - Pytorch 2.8.0+cpu
69
- - Datasets 4.2.0
70
- - Tokenizers 0.22.1
 
1
+ ---
2
+ library_name: peft
3
+ base_model: owenMills/bart-pas-model
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - rouge
8
+ model-index:
9
+ - name: bart-pas-lora
10
+ results: []
11
+ ---
 
 
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
  should probably proofread and complete it, then remove this comment. -->
 
17
 
18
  This model is a fine-tuned version of [owenMills/bart-pas-model](https://huggingface.co/owenMills/bart-pas-model) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 2.6566
21
+ - Rouge1: 47.54
22
+ - Rouge2: 39.76
23
+ - Rougel: 44.21
24
+ - Rougelsum: 44.06
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 0.0001
44
  - train_batch_size: 2
45
  - eval_batch_size: 2
46
  - seed: 42
47
  - gradient_accumulation_steps: 8
48
  - total_train_batch_size: 16
49
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
50
  - lr_scheduler_type: linear
51
  - lr_scheduler_warmup_ratio: 0.05
52
+ - num_epochs: 10
53
+ - mixed_precision_training: Native AMP
54
  - label_smoothing_factor: 0.1
55
 
56
  ### Training results
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
59
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
60
+ | 0.6531 | 1.0 | 12 | 2.6566 | 47.54 | 39.76 | 44.21 | 44.06 |
61
+ | 0.3393 | 2.0 | 24 | 2.6433 | 43.83 | 36.99 | 41.69 | 41.9 |
62
+ | 0.6332 | 3.0 | 36 | 2.6427 | 44.43 | 37.55 | 42.21 | 42.11 |
63
+ | 0.5081 | 4.0 | 48 | 2.6570 | 41.08 | 33.84 | 38.08 | 38.37 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - PEFT 0.13.2
69
+ - Transformers 4.49.0
70
+ - Pytorch 2.5.1+cu124
71
+ - Datasets 3.3.2
72
+ - Tokenizers 0.21.4
adapter_config.json CHANGED
@@ -1,39 +1,31 @@
1
- {
2
- "alpha_pattern": {},
3
- "auto_mapping": null,
4
- "base_model_name_or_path": "owenMills/bart-pas-model",
5
- "bias": "none",
6
- "corda_config": null,
7
- "eva_config": null,
8
- "exclude_modules": null,
9
- "fan_in_fan_out": false,
10
- "inference_mode": true,
11
- "init_lora_weights": true,
12
- "layer_replication": null,
13
- "layers_pattern": null,
14
- "layers_to_transform": null,
15
- "loftq_config": {},
16
- "lora_alpha": 16,
17
- "lora_bias": false,
18
- "lora_dropout": 0.05,
19
- "megatron_config": null,
20
- "megatron_core": "megatron.core",
21
- "modules_to_save": null,
22
- "peft_type": "LORA",
23
- "qalora_group_size": 16,
24
- "r": 8,
25
- "rank_pattern": {},
26
- "revision": null,
27
- "target_modules": [
28
- "k_proj",
29
- "q_proj",
30
- "v_proj",
31
- "out_proj"
32
- ],
33
- "target_parameters": null,
34
- "task_type": "SEQ_2_SEQ_LM",
35
- "trainable_token_indices": null,
36
- "use_dora": false,
37
- "use_qalora": false,
38
- "use_rslora": false
39
  }
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "owenMills/bart-pas-model",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 16,
14
+ "lora_dropout": 0.05,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "v_proj",
24
+ "out_proj",
25
+ "k_proj",
26
+ "q_proj"
27
+ ],
28
+ "task_type": "SEQ_2_SEQ_LM",
29
+ "use_dora": false,
30
+ "use_rslora": false
 
 
 
 
 
 
 
 
31
  }
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:46dd291dd9aff30ee62021f617563be413d61b36e57540c40218b43995d62155
3
  size 3558984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:102bc46c87182e794cf7ccee1e84b51d1369888d6e6476457e6a3114d6eee9da
3
  size 3558984
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_loss": 2.656590223312378,
3
+ "eval_rouge1": 47.54,
4
+ "eval_rouge2": 39.76,
5
+ "eval_rougeL": 44.21,
6
+ "eval_rougeLsum": 44.06,
7
+ "eval_runtime": 5.9984,
8
+ "eval_samples_per_second": 3.334,
9
+ "eval_steps_per_second": 1.667,
10
+ "epoch": 4.0
11
+ }
special_tokens_map.json CHANGED
@@ -1,51 +1,51 @@
1
- {
2
- "bos_token": {
3
- "content": "<s>",
4
- "lstrip": false,
5
- "normalized": true,
6
- "rstrip": false,
7
- "single_word": false
8
- },
9
- "cls_token": {
10
- "content": "<s>",
11
- "lstrip": false,
12
- "normalized": true,
13
- "rstrip": false,
14
- "single_word": false
15
- },
16
- "eos_token": {
17
- "content": "</s>",
18
- "lstrip": false,
19
- "normalized": true,
20
- "rstrip": false,
21
- "single_word": false
22
- },
23
- "mask_token": {
24
- "content": "<mask>",
25
- "lstrip": true,
26
- "normalized": true,
27
- "rstrip": false,
28
- "single_word": false
29
- },
30
- "pad_token": {
31
- "content": "<pad>",
32
- "lstrip": false,
33
- "normalized": true,
34
- "rstrip": false,
35
- "single_word": false
36
- },
37
- "sep_token": {
38
- "content": "</s>",
39
- "lstrip": false,
40
- "normalized": true,
41
- "rstrip": false,
42
- "single_word": false
43
- },
44
- "unk_token": {
45
- "content": "<unk>",
46
- "lstrip": false,
47
- "normalized": true,
48
- "rstrip": false,
49
- "single_word": false
50
- }
51
- }
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json CHANGED
@@ -2,7 +2,7 @@
2
  "version": "1.0",
3
  "truncation": {
4
  "direction": "Right",
5
- "max_length": 256,
6
  "strategy": "LongestFirst",
7
  "stride": 0
8
  },
 
2
  "version": "1.0",
3
  "truncation": {
4
  "direction": "Right",
5
+ "max_length": 192,
6
  "strategy": "LongestFirst",
7
  "stride": 0
8
  },
tokenizer_config.json CHANGED
@@ -1,62 +1,62 @@
1
- {
2
- "add_prefix_space": false,
3
- "added_tokens_decoder": {
4
- "0": {
5
- "content": "<s>",
6
- "lstrip": false,
7
- "normalized": true,
8
- "rstrip": false,
9
- "single_word": false,
10
- "special": true
11
- },
12
- "1": {
13
- "content": "<pad>",
14
- "lstrip": false,
15
- "normalized": true,
16
- "rstrip": false,
17
- "single_word": false,
18
- "special": true
19
- },
20
- "2": {
21
- "content": "</s>",
22
- "lstrip": false,
23
- "normalized": true,
24
- "rstrip": false,
25
- "single_word": false,
26
- "special": true
27
- },
28
- "3": {
29
- "content": "<unk>",
30
- "lstrip": false,
31
- "normalized": true,
32
- "rstrip": false,
33
- "single_word": false,
34
- "special": true
35
- },
36
- "50264": {
37
- "content": "<mask>",
38
- "lstrip": true,
39
- "normalized": true,
40
- "rstrip": false,
41
- "single_word": false,
42
- "special": true
43
- }
44
- },
45
- "bos_token": "<s>",
46
- "clean_up_tokenization_spaces": false,
47
- "cls_token": "<s>",
48
- "eos_token": "</s>",
49
- "errors": "replace",
50
- "extra_special_tokens": {},
51
- "mask_token": "<mask>",
52
- "max_length": 256,
53
- "model_max_length": 1000000000000000019884624838656,
54
- "pad_token": "<pad>",
55
- "sep_token": "</s>",
56
- "stride": 0,
57
- "tokenizer_class": "BartTokenizer",
58
- "trim_offsets": true,
59
- "truncation_side": "right",
60
- "truncation_strategy": "longest_first",
61
- "unk_token": "<unk>"
62
- }
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "extra_special_tokens": {},
51
+ "mask_token": "<mask>",
52
+ "max_length": 256,
53
+ "model_max_length": 1000000000000000019884624838656,
54
+ "pad_token": "<pad>",
55
+ "sep_token": "</s>",
56
+ "stride": 0,
57
+ "tokenizer_class": "BartTokenizer",
58
+ "trim_offsets": true,
59
+ "truncation_side": "right",
60
+ "truncation_strategy": "longest_first",
61
+ "unk_token": "<unk>"
62
+ }
train_results.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "train_loss": null,
3
+ "epoch": 4.0,
4
+ "best_model_checkpoint": "/opt/ml/output/data/checkpoint-12"
5
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:88605e9b4b9a12758e996712f8e0760d330610b609bae362b1057908e4f85e1c
3
- size 5969
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5761eec7e20fbda4837ae4eaa526fa1655ca941c29bb05f0a752453969c394d7
3
+ size 5496