zbigi commited on
Commit
078d448
1 Parent(s): b387f40

End of training

Browse files
README.md CHANGED
@@ -2,8 +2,6 @@
2
  base_model: facebook/bart-base
3
  library_name: peft
4
  license: apache-2.0
5
- metrics:
6
- - rouge
7
  tags:
8
  - generated_from_trainer
9
  model-index:
@@ -16,14 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # bart-base-summarization-medical-45
18
 
19
- This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on the None dataset.
20
- It achieves the following results on the evaluation set:
21
- - Loss: 2.1275
22
- - Rouge1: 0.4196
23
- - Rouge2: 0.2244
24
- - Rougel: 0.3562
25
- - Rougelsum: 0.3564
26
- - Gen Len: 18.338
27
 
28
  ## Model description
29
 
@@ -50,25 +41,13 @@ The following hyperparameters were used during training:
50
  - total_train_batch_size: 16
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
  - lr_scheduler_type: linear
53
- - num_epochs: 6
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
- | 2.6973 | 1.0 | 1250 | 2.1978 | 0.4152 | 0.2215 | 0.3551 | 0.3555 | 17.896 |
61
- | 2.6015 | 2.0 | 2500 | 2.1713 | 0.4141 | 0.2205 | 0.3511 | 0.3508 | 18.338 |
62
- | 2.5775 | 3.0 | 3750 | 2.1466 | 0.4165 | 0.2208 | 0.3514 | 0.3515 | 18.41 |
63
- | 2.5536 | 4.0 | 5000 | 2.1345 | 0.4214 | 0.2241 | 0.358 | 0.3579 | 18.38 |
64
- | 2.5322 | 5.0 | 6250 | 2.1279 | 0.4226 | 0.2257 | 0.3587 | 0.3587 | 18.447 |
65
- | 2.5289 | 6.0 | 7500 | 2.1275 | 0.4196 | 0.2244 | 0.3562 | 0.3564 | 18.338 |
66
-
67
-
68
  ### Framework versions
69
 
70
- - PEFT 0.12.0
71
- - Transformers 4.42.4
72
- - Pytorch 2.3.1+cu121
73
- - Datasets 2.20.0
74
  - Tokenizers 0.19.1
 
2
  base_model: facebook/bart-base
3
  library_name: peft
4
  license: apache-2.0
 
 
5
  tags:
6
  - generated_from_trainer
7
  model-index:
 
14
 
15
  # bart-base-summarization-medical-45
16
 
17
+ This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on an unknown dataset.
 
 
 
 
 
 
 
18
 
19
  ## Model description
20
 
 
41
  - total_train_batch_size: 16
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
+ - num_epochs: 10
45
  - mixed_precision_training: Native AMP
46
 
 
 
 
 
 
 
 
 
 
 
 
 
47
  ### Framework versions
48
 
49
+ - PEFT 0.13.2
50
+ - Transformers 4.44.2
51
+ - Pytorch 2.5.0+cu121
52
+ - Datasets 3.1.0
53
  - Tokenizers 0.19.1
adapter_config.json CHANGED
@@ -10,18 +10,20 @@
10
  "layers_pattern": null,
11
  "layers_to_transform": null,
12
  "loftq_config": {},
13
- "lora_alpha": 32,
14
  "lora_dropout": 0.01,
15
  "megatron_config": null,
16
  "megatron_core": "megatron.core",
17
  "modules_to_save": null,
18
  "peft_type": "LORA",
19
- "r": 4,
20
  "rank_pattern": {},
21
  "revision": null,
22
  "target_modules": [
 
 
23
  "v_proj",
24
- "q_proj"
25
  ],
26
  "task_type": "SEQ_2_SEQ_LM",
27
  "use_dora": false,
 
10
  "layers_pattern": null,
11
  "layers_to_transform": null,
12
  "loftq_config": {},
13
+ "lora_alpha": 128,
14
  "lora_dropout": 0.01,
15
  "megatron_config": null,
16
  "megatron_core": "megatron.core",
17
  "modules_to_save": null,
18
  "peft_type": "LORA",
19
+ "r": 64,
20
  "rank_pattern": {},
21
  "revision": null,
22
  "target_modules": [
23
+ "q_proj",
24
+ "o_proj",
25
  "v_proj",
26
+ "k_proj"
27
  ],
28
  "task_type": "SEQ_2_SEQ_LM",
29
  "use_dora": false,
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2638aa44ac9544639bfbde956f4a051be5e4c908e70534e40f38788f4ed3babd
3
- size 894624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d524cadb866845135d808fbbb7b8cc01225cb79af16f40c52296924a866fe273
3
+ size 21248928
runs/Nov11_09-32-20_ae1726d9306b/events.out.tfevents.1731318527.ae1726d9306b.78712.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58ed4dc70cf5fc1065c4a090eba941139398da06e4cd4feafabd153afd45e401
3
+ size 1270
tokenizer.json CHANGED
@@ -1,11 +1,6 @@
1
  {
2
  "version": "1.0",
3
- "truncation": {
4
- "direction": "Right",
5
- "max_length": 128,
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
  {
tokenizer_config.json CHANGED
@@ -50,8 +50,10 @@
50
  "mask_token": "<mask>",
51
  "model_max_length": 1000000000000000019884624838656,
52
  "pad_token": "<pad>",
 
53
  "sep_token": "</s>",
54
  "tokenizer_class": "BartTokenizer",
55
  "trim_offsets": true,
 
56
  "unk_token": "<unk>"
57
  }
 
50
  "mask_token": "<mask>",
51
  "model_max_length": 1000000000000000019884624838656,
52
  "pad_token": "<pad>",
53
+ "padding": true,
54
  "sep_token": "</s>",
55
  "tokenizer_class": "BartTokenizer",
56
  "trim_offsets": true,
57
+ "truncation": true,
58
  "unk_token": "<unk>"
59
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8cdca8155007849cb51f1d981ecad3c9476b6c9ef031e59bdc257ec6f09e59b
3
- size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:441a06469c1a3a5e9e294be775687802c0c5f99d3a09020232976dfcbb693195
3
+ size 5432