Instructions to use taiypeo/bart-large-wiki-doc-full with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use taiypeo/bart-large-wiki-doc-full with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("taiypeo/bart-large-wiki-doc-full") model = AutoModelForSeq2SeqLM.from_pretrained("taiypeo/bart-large-wiki-doc-full") - Notebooks
- Google Colab
- Kaggle
Model save
Browse files- README.md +91 -0
- merges.txt +0 -0
- model.safetensors +1 -1
- special_tokens_map.json +15 -0
- tokenizer.json +0 -0
- tokenizer_config.json +58 -0
- vocab.json +0 -0
README.md
ADDED
|
@@ -0,0 +1,91 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: transformers
|
| 3 |
+
license: apache-2.0
|
| 4 |
+
base_model: facebook/bart-large
|
| 5 |
+
tags:
|
| 6 |
+
- generated_from_trainer
|
| 7 |
+
metrics:
|
| 8 |
+
- bleu
|
| 9 |
+
model-index:
|
| 10 |
+
- name: bart-large-wiki-doc-full
|
| 11 |
+
results: []
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 15 |
+
should probably proofread and complete it, then remove this comment. -->
|
| 16 |
+
|
| 17 |
+
# bart-large-wiki-doc-full
|
| 18 |
+
|
| 19 |
+
This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
|
| 20 |
+
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 4.9843
|
| 22 |
+
- Sari: 51.0933
|
| 23 |
+
- Sari Add: 12.516
|
| 24 |
+
- Sari Keep: 45.354
|
| 25 |
+
- Sari Del: 95.4099
|
| 26 |
+
- Fkgl: 6.642
|
| 27 |
+
- Bleu: 19.5688
|
| 28 |
+
- D Sari: 0.4555
|
| 29 |
+
- D Sari Keep: 0.3857
|
| 30 |
+
- D Sari Del: 0.8046
|
| 31 |
+
- D Sari Add: 0.1763
|
| 32 |
+
|
| 33 |
+
## Model description
|
| 34 |
+
|
| 35 |
+
More information needed
|
| 36 |
+
|
| 37 |
+
## Intended uses & limitations
|
| 38 |
+
|
| 39 |
+
More information needed
|
| 40 |
+
|
| 41 |
+
## Training and evaluation data
|
| 42 |
+
|
| 43 |
+
More information needed
|
| 44 |
+
|
| 45 |
+
## Training procedure
|
| 46 |
+
|
| 47 |
+
### Training hyperparameters
|
| 48 |
+
|
| 49 |
+
The following hyperparameters were used during training:
|
| 50 |
+
- learning_rate: 5e-05
|
| 51 |
+
- train_batch_size: 2
|
| 52 |
+
- eval_batch_size: 2
|
| 53 |
+
- seed: 42
|
| 54 |
+
- gradient_accumulation_steps: 4
|
| 55 |
+
- total_train_batch_size: 8
|
| 56 |
+
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 57 |
+
- lr_scheduler_type: linear
|
| 58 |
+
- lr_scheduler_warmup_ratio: 0.06
|
| 59 |
+
- num_epochs: 15
|
| 60 |
+
- label_smoothing_factor: 0.3
|
| 61 |
+
|
| 62 |
+
### Training results
|
| 63 |
+
|
| 64 |
+
| Training Loss | Epoch | Step | Validation Loss | Sari | Sari Add | Sari Keep | Sari Del | Fkgl | Bleu | D Sari | D Sari Keep | D Sari Del | D Sari Add |
|
| 65 |
+
|:-------------:|:------:|:----:|:---------------:|:-------:|:--------:|:---------:|:--------:|:------:|:-------:|:------:|:-----------:|:----------:|:----------:|
|
| 66 |
+
| 5.6237 | 0.2862 | 500 | 5.1875 | 47.043 | 5.9566 | 40.0258 | 95.1466 | 7.2471 | 12.9924 | 0.4057 | 0.3169 | 0.7965 | 0.1038 |
|
| 67 |
+
| 5.0682 | 0.5725 | 1000 | 5.1042 | 48.192 | 9.0538 | 40.5192 | 95.0031 | 7.2139 | 15.7017 | 0.4288 | 0.3477 | 0.8055 | 0.1332 |
|
| 68 |
+
| 5.0363 | 0.8587 | 1500 | 5.0920 | 48.4536 | 9.2349 | 41.6306 | 94.4954 | 7.6174 | 20.3138 | 0.4245 | 0.3505 | 0.7871 | 0.1358 |
|
| 69 |
+
| 4.9284 | 1.1448 | 2000 | 5.0709 | 49.1161 | 9.9088 | 42.5012 | 94.9385 | 7.3117 | 18.5046 | 0.4339 | 0.3599 | 0.7961 | 0.1455 |
|
| 70 |
+
| 4.8644 | 1.4311 | 2500 | 5.0343 | 49.604 | 10.9617 | 42.8337 | 95.0164 | 7.1277 | 18.8397 | 0.4403 | 0.3681 | 0.7991 | 0.1536 |
|
| 71 |
+
| 4.8532 | 1.7173 | 3000 | 5.0192 | 49.3028 | 10.3908 | 42.1003 | 95.4172 | 6.9579 | 14.4955 | 0.4389 | 0.3588 | 0.8116 | 0.1462 |
|
| 72 |
+
| 4.8343 | 2.0034 | 3500 | 5.0058 | 49.2487 | 11.0341 | 42.1651 | 94.5467 | 7.2777 | 21.4311 | 0.4401 | 0.3693 | 0.7926 | 0.1583 |
|
| 73 |
+
| 4.719 | 2.2897 | 4000 | 5.0043 | 49.8632 | 11.2301 | 42.9018 | 95.4578 | 6.5783 | 15.364 | 0.4509 | 0.3735 | 0.8138 | 0.1654 |
|
| 74 |
+
| 4.6898 | 2.5759 | 4500 | 4.9904 | 49.9624 | 11.3392 | 43.3503 | 95.1978 | 6.9214 | 18.1433 | 0.449 | 0.3764 | 0.8068 | 0.1637 |
|
| 75 |
+
| 4.7061 | 2.8622 | 5000 | 4.9840 | 49.9367 | 11.8606 | 42.6481 | 95.3013 | 6.9934 | 16.9996 | 0.4516 | 0.3748 | 0.8158 | 0.1643 |
|
| 76 |
+
| 4.6392 | 3.1483 | 5500 | 4.9898 | 50.5502 | 11.928 | 44.3611 | 95.3614 | 6.9017 | 18.7169 | 0.4523 | 0.3813 | 0.8091 | 0.1665 |
|
| 77 |
+
| 4.578 | 3.4345 | 6000 | 4.9893 | 50.39 | 12.0634 | 43.6891 | 95.4175 | 6.8384 | 17.0137 | 0.4567 | 0.3837 | 0.8154 | 0.171 |
|
| 78 |
+
| 4.5981 | 3.7208 | 6500 | 4.9843 | 51.0933 | 12.516 | 45.354 | 95.4099 | 6.642 | 19.5688 | 0.4555 | 0.3857 | 0.8046 | 0.1763 |
|
| 79 |
+
| 4.5826 | 4.0069 | 7000 | 4.9818 | 50.788 | 12.8273 | 44.4249 | 95.1118 | 6.6555 | 21.2107 | 0.4594 | 0.3942 | 0.8049 | 0.1791 |
|
| 80 |
+
| 4.4927 | 4.2931 | 7500 | 5.0009 | 50.4098 | 12.2269 | 43.9055 | 95.0968 | 6.754 | 20.3637 | 0.4602 | 0.3948 | 0.8096 | 0.1763 |
|
| 81 |
+
| 4.5009 | 4.5794 | 8000 | 4.9908 | 50.7179 | 12.8769 | 43.783 | 95.4938 | 6.5702 | 17.565 | 0.4601 | 0.3863 | 0.8152 | 0.1787 |
|
| 82 |
+
| 4.5128 | 4.8656 | 8500 | 4.9830 | 50.7341 | 12.5935 | 44.2741 | 95.3346 | 6.7056 | 18.9024 | 0.4588 | 0.3905 | 0.8076 | 0.1783 |
|
| 83 |
+
| 4.457 | 5.1517 | 9000 | 5.0095 | 50.8947 | 12.7442 | 44.829 | 95.111 | 6.8278 | 21.5163 | 0.462 | 0.3987 | 0.8062 | 0.181 |
|
| 84 |
+
|
| 85 |
+
|
| 86 |
+
### Framework versions
|
| 87 |
+
|
| 88 |
+
- Transformers 4.57.3
|
| 89 |
+
- Pytorch 2.9.1+cu128
|
| 90 |
+
- Datasets 3.6.0
|
| 91 |
+
- Tokenizers 0.22.1
|
merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1625426996
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9b1c15626f3fffc5c3693ab6b2545bfb46b4fef1e54c36a51da1b612a2cc585e
|
| 3 |
size 1625426996
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": "<s>",
|
| 3 |
+
"cls_token": "<s>",
|
| 4 |
+
"eos_token": "</s>",
|
| 5 |
+
"mask_token": {
|
| 6 |
+
"content": "<mask>",
|
| 7 |
+
"lstrip": true,
|
| 8 |
+
"normalized": true,
|
| 9 |
+
"rstrip": false,
|
| 10 |
+
"single_word": false
|
| 11 |
+
},
|
| 12 |
+
"pad_token": "<pad>",
|
| 13 |
+
"sep_token": "</s>",
|
| 14 |
+
"unk_token": "<unk>"
|
| 15 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
"model_max_length": 1024,
|
| 53 |
+
"pad_token": "<pad>",
|
| 54 |
+
"sep_token": "</s>",
|
| 55 |
+
"tokenizer_class": "BartTokenizer",
|
| 56 |
+
"trim_offsets": true,
|
| 57 |
+
"unk_token": "<unk>"
|
| 58 |
+
}
|
vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|