zbigi commited on
Commit
a24c528
1 Parent(s): 1518d19

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-43
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.1317
22
- - Rouge1: 0.4198
23
- - Rouge2: 0.2252
24
- - Rougel: 0.3557
25
- - Rougelsum: 0.3554
26
- - Gen Len: 18.34
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.7087 | 1.0 | 1250 | 2.2005 | 0.4151 | 0.2224 | 0.3555 | 0.3553 | 18.135 |
61
- | 2.615 | 2.0 | 2500 | 2.1680 | 0.4145 | 0.2228 | 0.3537 | 0.3534 | 17.898 |
62
- | 2.571 | 3.0 | 3750 | 2.1526 | 0.417 | 0.2224 | 0.3525 | 0.3522 | 18.122 |
63
- | 2.5533 | 4.0 | 5000 | 2.1433 | 0.4199 | 0.2239 | 0.3558 | 0.3555 | 18.241 |
64
- | 2.5482 | 5.0 | 6250 | 2.1344 | 0.4222 | 0.226 | 0.3562 | 0.356 | 18.293 |
65
- | 2.5365 | 6.0 | 7500 | 2.1317 | 0.4198 | 0.2252 | 0.3557 | 0.3554 | 18.34 |
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-43
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
  "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:8e9b32f160fa0ca74751defa95e9a8f7aa762b6d6c17a51687898e3ef805b545
3
- size 894624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3387c8a9d7a5f8ccf058428830e0938b2f511d5e3c7018e4a78ac94c66bedd54
3
+ size 21248928
runs/Nov10_23-45-17_ae1726d9306b/events.out.tfevents.1731283304.ae1726d9306b.78712.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759bb56719ebce086483e47b21284c292f7ff3afe6768f583f7c912ee25ed7cc
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:259de4c565f43e9912b84df74079ba9964058c77db7e1b7d67580340b2c68905
3
- size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d11af2468743fafc66c2f9e19c95c0719d1345bbd8839742062ec1e09e3d8b62
3
+ size 5432