zbigi commited on
Commit
bbb451f
1 Parent(s): 2de6cea

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-46
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.1274
22
- - Rouge1: 0.4211
23
- - Rouge2: 0.2237
24
- - Rougel: 0.3559
25
- - Rougelsum: 0.356
26
- - Gen Len: 18.296
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.7041 | 1.0 | 1250 | 2.2079 | 0.4109 | 0.2179 | 0.35 | 0.3501 | 18.052 |
61
- | 2.6134 | 2.0 | 2500 | 2.1641 | 0.4151 | 0.2203 | 0.3517 | 0.3518 | 18.103 |
62
- | 2.5664 | 3.0 | 3750 | 2.1450 | 0.4163 | 0.22 | 0.353 | 0.3529 | 18.051 |
63
- | 2.5501 | 4.0 | 5000 | 2.1328 | 0.4191 | 0.2217 | 0.353 | 0.3528 | 18.228 |
64
- | 2.535 | 5.0 | 6250 | 2.1268 | 0.4221 | 0.2253 | 0.3569 | 0.3568 | 18.235 |
65
- | 2.5102 | 6.0 | 7500 | 2.1274 | 0.4211 | 0.2237 | 0.3559 | 0.356 | 18.296 |
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-46
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
  "q_proj",
24
- "v_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
+ "v_proj",
24
  "q_proj",
25
+ "k_proj",
26
+ "o_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:973d5c2b3f8b0397491a7a0c89c5c317efa1cf4298167b232d1fda9daaa6c936
3
- size 894624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe692703cd19414e7c350f96e00fc770910e654d57263c6b87fe9c7a96857b0c
3
+ size 21248928
runs/Nov12_04-43-48_c922dae45a8f/events.out.tfevents.1731387630.c922dae45a8f.5756.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc85f32f951f9a1b76950473365e6fe6f663c5d9dac86f175810b0df56473ff7
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:5fb31b410a013e9683d9e87f988ed3ef8dbf6655e6094490b7276ad3a61c77ed
3
- size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07e177ec0d8613fa321aec37ecb3ec0eeec5c0a07a25171d3c2c7dc55e4bc8d0
3
+ size 5432