ai-forever commited on
Commit
6af51a0
·
1 Parent(s): cc6e929

Upload pipeline weights

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +4 -0
  2. bart_response_generator/README.md +66 -0
  3. bart_response_generator/added_tokens.json +4 -0
  4. bart_response_generator/all_results.json +18 -0
  5. bart_response_generator/config.json +74 -0
  6. bart_response_generator/eval_results.json +13 -0
  7. bart_response_generator/generation_config.json +12 -0
  8. bart_response_generator/merges.txt +0 -0
  9. bart_response_generator/pytorch_model.bin +3 -0
  10. bart_response_generator/special_tokens_map.json +18 -0
  11. bart_response_generator/tokenizer.json +0 -0
  12. bart_response_generator/tokenizer_config.json +15 -0
  13. bart_response_generator/train_results.json +8 -0
  14. bart_response_generator/trainer_state.json +101 -0
  15. bart_response_generator/training_args.bin +3 -0
  16. bart_response_generator/vocab.json +0 -0
  17. classifier_disco.pth +3 -0
  18. convokit_50_model/checkpoint +2 -0
  19. convokit_50_model/checkpoint-00000050.data-00000-of-00001 +3 -0
  20. convokit_50_model/checkpoint-00000050.index +0 -0
  21. convokit_50_model/checkpoint-00000050.meta +3 -0
  22. convokit_dials_train.json +3 -0
  23. deberta_qa/all_results.json +8 -0
  24. deberta_qa/config.json +41 -0
  25. deberta_qa/pytorch_model.bin +3 -0
  26. deberta_qa/special_tokens_map.json +9 -0
  27. deberta_qa/tokenizer.json +0 -0
  28. deberta_qa/tokenizer_config.json +17 -0
  29. deberta_qa/train_results.json +8 -0
  30. deberta_qa/trainer_state.json +1009 -0
  31. deberta_qa/training_args.bin +3 -0
  32. distilbart_summarizer/README.md +66 -0
  33. distilbart_summarizer/all_results.json +18 -0
  34. distilbart_summarizer/config.json +77 -0
  35. distilbart_summarizer/eval_results.json +13 -0
  36. distilbart_summarizer/generation_config.json +15 -0
  37. distilbart_summarizer/merges.txt +0 -0
  38. distilbart_summarizer/pytorch_model.bin +3 -0
  39. distilbart_summarizer/special_tokens_map.json +51 -0
  40. distilbart_summarizer/tokenizer.json +0 -0
  41. distilbart_summarizer/tokenizer_config.json +64 -0
  42. distilbart_summarizer/train_results.json +8 -0
  43. distilbart_summarizer/trainer_state.json +101 -0
  44. distilbart_summarizer/training_args.bin +3 -0
  45. distilbart_summarizer/vocab.json +0 -0
  46. glove.6B.100d.txt +3 -0
  47. sentence_bert_disco/.DS_Store +0 -0
  48. sentence_bert_disco/1_Pooling/config.json +7 -0
  49. sentence_bert_disco/README.md +125 -0
  50. sentence_bert_disco/config.json +26 -0
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ convokit_50_model/checkpoint-00000050.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
37
+ convokit_50_model/checkpoint-00000050.meta filter=lfs diff=lfs merge=lfs -text
38
+ convokit_dials_train.json filter=lfs diff=lfs merge=lfs -text
39
+ glove.6B.100d.txt filter=lfs diff=lfs merge=lfs -text
bart_response_generator/README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: BLUE_bart_large_5ep_5e_5
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # BLUE_bart_large_5ep_5e_5
16
+
17
+ This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.0958
20
+ - Rouge1: 67.9321
21
+ - Rouge2: 54.9471
22
+ - Rougel: 60.922
23
+ - Rougelsum: 63.8581
24
+ - Gen Len: 51.8306
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - learning_rate: 5e-05
44
+ - train_batch_size: 20
45
+ - eval_batch_size: 20
46
+ - seed: 42
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: linear
49
+ - num_epochs: 5.0
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
54
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
55
+ | 1.2388 | 1.07 | 2000 | 1.1304 | 66.5164 | 54.1964 | 60.1255 | 62.6329 | 48.5373 |
56
+ | 0.9911 | 2.14 | 4000 | 1.0997 | 66.7923 | 54.4921 | 60.3864 | 63.0007 | 48.2194 |
57
+ | 0.8473 | 3.2 | 6000 | 1.1019 | 68.1867 | 55.4038 | 61.5061 | 64.2051 | 51.4172 |
58
+ | 0.7315 | 4.27 | 8000 | 1.1118 | 67.961 | 55.3863 | 61.4022 | 64.1305 | 50.4269 |
59
+
60
+
61
+ ### Framework versions
62
+
63
+ - Transformers 4.28.1
64
+ - Pytorch 1.8.1+cu111
65
+ - Datasets 2.11.0
66
+ - Tokenizers 0.13.3
bart_response_generator/added_tokens.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "<SEP>": 50265,
3
+ "<UTTERSEP>": 50266
4
+ }
bart_response_generator/all_results.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_gen_len": 51.830597014925374,
4
+ "eval_loss": 1.0958350896835327,
5
+ "eval_rouge1": 67.9321,
6
+ "eval_rouge2": 54.9471,
7
+ "eval_rougeL": 60.922,
8
+ "eval_rougeLsum": 63.8581,
9
+ "eval_runtime": 265.3264,
10
+ "eval_samples": 1340,
11
+ "eval_samples_per_second": 5.05,
12
+ "eval_steps_per_second": 0.253,
13
+ "train_loss": 0.9121758847458122,
14
+ "train_runtime": 3599.8478,
15
+ "train_samples": 37460,
16
+ "train_samples_per_second": 52.03,
17
+ "train_steps_per_second": 2.601
18
+ }
bart_response_generator/config.json ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/bart-large",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "BartForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 0,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 1024,
15
+ "decoder_attention_heads": 16,
16
+ "decoder_ffn_dim": 4096,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 12,
19
+ "decoder_start_token_id": 2,
20
+ "dropout": 0.1,
21
+ "early_stopping": true,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 12,
26
+ "eos_token_id": 2,
27
+ "forced_bos_token_id": 0,
28
+ "forced_eos_token_id": 2,
29
+ "gradient_checkpointing": false,
30
+ "id2label": {
31
+ "0": "LABEL_0",
32
+ "1": "LABEL_1",
33
+ "2": "LABEL_2"
34
+ },
35
+ "init_std": 0.02,
36
+ "is_encoder_decoder": true,
37
+ "label2id": {
38
+ "LABEL_0": 0,
39
+ "LABEL_1": 1,
40
+ "LABEL_2": 2
41
+ },
42
+ "max_position_embeddings": 1024,
43
+ "model_type": "bart",
44
+ "no_repeat_ngram_size": 3,
45
+ "normalize_before": false,
46
+ "num_beams": 4,
47
+ "num_hidden_layers": 12,
48
+ "pad_token_id": 1,
49
+ "scale_embedding": false,
50
+ "task_specific_params": {
51
+ "summarization": {
52
+ "length_penalty": 1.0,
53
+ "max_length": 128,
54
+ "min_length": 12,
55
+ "num_beams": 4
56
+ },
57
+ "summarization_cnn": {
58
+ "length_penalty": 2.0,
59
+ "max_length": 142,
60
+ "min_length": 56,
61
+ "num_beams": 4
62
+ },
63
+ "summarization_xsum": {
64
+ "length_penalty": 1.0,
65
+ "max_length": 62,
66
+ "min_length": 11,
67
+ "num_beams": 6
68
+ }
69
+ },
70
+ "torch_dtype": "float32",
71
+ "transformers_version": "4.28.1",
72
+ "use_cache": true,
73
+ "vocab_size": 50267
74
+ }
bart_response_generator/eval_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_gen_len": 51.830597014925374,
4
+ "eval_loss": 1.0958350896835327,
5
+ "eval_rouge1": 67.9321,
6
+ "eval_rouge2": 54.9471,
7
+ "eval_rougeL": 60.922,
8
+ "eval_rougeLsum": 63.8581,
9
+ "eval_runtime": 265.3264,
10
+ "eval_samples": 1340,
11
+ "eval_samples_per_second": 5.05,
12
+ "eval_steps_per_second": 0.253
13
+ }
bart_response_generator/generation_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 2,
4
+ "early_stopping": true,
5
+ "eos_token_id": 2,
6
+ "forced_bos_token_id": 0,
7
+ "forced_eos_token_id": 2,
8
+ "no_repeat_ngram_size": 3,
9
+ "num_beams": 4,
10
+ "pad_token_id": 1,
11
+ "transformers_version": "4.28.1"
12
+ }
bart_response_generator/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
bart_response_generator/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08509447caa443e6bf3b9a4848bbe1877bcf984c42b95dc5a0bf5712f97d7be1
3
+ size 1625558063
bart_response_generator/special_tokens_map.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<UTTERSEP>"
4
+ ],
5
+ "bos_token": "<s>",
6
+ "cls_token": "<s>",
7
+ "eos_token": "</s>",
8
+ "mask_token": {
9
+ "content": "<mask>",
10
+ "lstrip": true,
11
+ "normalized": false,
12
+ "rstrip": false,
13
+ "single_word": false
14
+ },
15
+ "pad_token": "<pad>",
16
+ "sep_token": "<SEP>",
17
+ "unk_token": "<unk>"
18
+ }
bart_response_generator/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
bart_response_generator/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<s>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "errors": "replace",
8
+ "mask_token": "<mask>",
9
+ "model_max_length": 1024,
10
+ "pad_token": "<pad>",
11
+ "sep_token": "</s>",
12
+ "tokenizer_class": "BartTokenizer",
13
+ "trim_offsets": true,
14
+ "unk_token": "<unk>"
15
+ }
bart_response_generator/train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "train_loss": 0.9121758847458122,
4
+ "train_runtime": 3599.8478,
5
+ "train_samples": 37460,
6
+ "train_samples_per_second": 52.03,
7
+ "train_steps_per_second": 2.601
8
+ }
bart_response_generator/trainer_state.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 5.0,
5
+ "global_step": 9365,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 1.07,
12
+ "learning_rate": 3.932194340630005e-05,
13
+ "loss": 1.2388,
14
+ "step": 2000
15
+ },
16
+ {
17
+ "epoch": 1.07,
18
+ "eval_gen_len": 48.53731343283582,
19
+ "eval_loss": 1.1303935050964355,
20
+ "eval_rouge1": 66.5164,
21
+ "eval_rouge2": 54.1964,
22
+ "eval_rougeL": 60.1255,
23
+ "eval_rougeLsum": 62.6329,
24
+ "eval_runtime": 248.0848,
25
+ "eval_samples_per_second": 5.401,
26
+ "eval_steps_per_second": 0.27,
27
+ "step": 2000
28
+ },
29
+ {
30
+ "epoch": 2.14,
31
+ "learning_rate": 2.8643886812600106e-05,
32
+ "loss": 0.9911,
33
+ "step": 4000
34
+ },
35
+ {
36
+ "epoch": 2.14,
37
+ "eval_gen_len": 48.21940298507463,
38
+ "eval_loss": 1.0996825695037842,
39
+ "eval_rouge1": 66.7923,
40
+ "eval_rouge2": 54.4921,
41
+ "eval_rougeL": 60.3864,
42
+ "eval_rougeLsum": 63.0007,
43
+ "eval_runtime": 245.1086,
44
+ "eval_samples_per_second": 5.467,
45
+ "eval_steps_per_second": 0.273,
46
+ "step": 4000
47
+ },
48
+ {
49
+ "epoch": 3.2,
50
+ "learning_rate": 1.7965830218900163e-05,
51
+ "loss": 0.8473,
52
+ "step": 6000
53
+ },
54
+ {
55
+ "epoch": 3.2,
56
+ "eval_gen_len": 51.417164179104475,
57
+ "eval_loss": 1.1019301414489746,
58
+ "eval_rouge1": 68.1867,
59
+ "eval_rouge2": 55.4038,
60
+ "eval_rougeL": 61.5061,
61
+ "eval_rougeLsum": 64.2051,
62
+ "eval_runtime": 255.7993,
63
+ "eval_samples_per_second": 5.238,
64
+ "eval_steps_per_second": 0.262,
65
+ "step": 6000
66
+ },
67
+ {
68
+ "epoch": 4.27,
69
+ "learning_rate": 7.287773625200214e-06,
70
+ "loss": 0.7315,
71
+ "step": 8000
72
+ },
73
+ {
74
+ "epoch": 4.27,
75
+ "eval_gen_len": 50.42686567164179,
76
+ "eval_loss": 1.111816167831421,
77
+ "eval_rouge1": 67.961,
78
+ "eval_rouge2": 55.3863,
79
+ "eval_rougeL": 61.4022,
80
+ "eval_rougeLsum": 64.1305,
81
+ "eval_runtime": 261.9465,
82
+ "eval_samples_per_second": 5.116,
83
+ "eval_steps_per_second": 0.256,
84
+ "step": 8000
85
+ },
86
+ {
87
+ "epoch": 5.0,
88
+ "step": 9365,
89
+ "total_flos": 1.1949135422717952e+17,
90
+ "train_loss": 0.9121758847458122,
91
+ "train_runtime": 3599.8478,
92
+ "train_samples_per_second": 52.03,
93
+ "train_steps_per_second": 2.601
94
+ }
95
+ ],
96
+ "max_steps": 9365,
97
+ "num_train_epochs": 5,
98
+ "total_flos": 1.1949135422717952e+17,
99
+ "trial_name": null,
100
+ "trial_params": null
101
+ }
bart_response_generator/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f94865fe6e34e476c79baddd0c4bd3cc8fcdcfadfa8b823585bc544ee66ed0a
3
+ size 3823
bart_response_generator/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
classifier_disco.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47fdafcea1a35894bbf2984f0c9d0059950929baec382776b95d19416a44e2c4
3
+ size 29095
convokit_50_model/checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model_checkpoint_path: "checkpoint-00000050"
2
+ all_model_checkpoint_paths: "checkpoint-00000050"
convokit_50_model/checkpoint-00000050.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7913fbe973da462ad5bd02abcbc86ad424c5a2507b729eabdf622813184f631d
3
+ size 64192566
convokit_50_model/checkpoint-00000050.index ADDED
Binary file (3.09 kB). View file
 
convokit_50_model/checkpoint-00000050.meta ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b28032d4f0b302ede82efd604bb410fd1453498227474b8f0ed0700c1e89584
3
+ size 98155604
convokit_dials_train.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66e1d3aad756d0d6e4d547ab7a48166722e020fd89b183ea67b7de0720167a17
3
+ size 33319326
deberta_qa/all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "train_loss": 0.5708848007960676,
4
+ "train_runtime": 17933.1446,
5
+ "train_samples": 164635,
6
+ "train_samples_per_second": 36.722,
7
+ "train_steps_per_second": 4.59
8
+ }
deberta_qa/config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "deepset/deberta-v3-base-squad2",
3
+ "architectures": [
4
+ "DebertaV2ForQuestionAnswering"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 768,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 3072,
12
+ "language": "english",
13
+ "layer_norm_eps": 1e-07,
14
+ "max_position_embeddings": 512,
15
+ "max_relative_positions": -1,
16
+ "model_type": "deberta-v2",
17
+ "name": "DebertaV2",
18
+ "norm_rel_ebd": "layer_norm",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 12,
21
+ "pad_token_id": 0,
22
+ "pooler_dropout": 0,
23
+ "pooler_hidden_act": "gelu",
24
+ "pooler_hidden_size": 768,
25
+ "pos_att_type": [
26
+ "p2c",
27
+ "c2p"
28
+ ],
29
+ "position_biased_input": false,
30
+ "position_buckets": 256,
31
+ "relative_attention": true,
32
+ "share_att_key": true,
33
+ "summary_activation": "tanh",
34
+ "summary_last_dropout": 0,
35
+ "summary_type": "first",
36
+ "summary_use_proj": false,
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.26.0",
39
+ "type_vocab_size": 0,
40
+ "vocab_size": 128100
41
+ }
deberta_qa/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8dd9732db877a6e31e2df7bc5e03c45d661fa303e17075e79e8ffccc2b0066c3
3
+ size 735405809
deberta_qa/special_tokens_map.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "[CLS]",
3
+ "cls_token": "[CLS]",
4
+ "eos_token": "[SEP]",
5
+ "mask_token": "[MASK]",
6
+ "pad_token": "[PAD]",
7
+ "sep_token": "[SEP]",
8
+ "unk_token": "[UNK]"
9
+ }
deberta_qa/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
deberta_qa/tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "[CLS]",
3
+ "cls_token": "[CLS]",
4
+ "do_lower_case": false,
5
+ "eos_token": "[SEP]",
6
+ "mask_token": "[MASK]",
7
+ "model_max_length": 512,
8
+ "name_or_path": "deepset/deberta-v3-base-squad2",
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "sp_model_kwargs": {},
12
+ "special_tokens_map_file": null,
13
+ "split_by_punct": false,
14
+ "tokenizer_class": "DebertaV2Tokenizer",
15
+ "unk_token": "[UNK]",
16
+ "vocab_type": "spm"
17
+ }
deberta_qa/train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "train_loss": 0.5708848007960676,
4
+ "train_runtime": 17933.1446,
5
+ "train_samples": 164635,
6
+ "train_samples_per_second": 36.722,
7
+ "train_steps_per_second": 4.59
8
+ }
deberta_qa/trainer_state.json ADDED
@@ -0,0 +1,1009 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.9999757039772588,
5
+ "global_step": 82316,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.02,
12
+ "learning_rate": 1.98785169347393e-05,
13
+ "loss": 1.0418,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.05,
18
+ "learning_rate": 1.9757033869478596e-05,
19
+ "loss": 0.8326,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.07,
24
+ "learning_rate": 1.9635550804217894e-05,
25
+ "loss": 0.8343,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.1,
30
+ "learning_rate": 1.951406773895719e-05,
31
+ "loss": 0.8015,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.12,
36
+ "learning_rate": 1.939258467369649e-05,
37
+ "loss": 0.7903,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.15,
42
+ "learning_rate": 1.9271101608435786e-05,
43
+ "loss": 0.7802,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.17,
48
+ "learning_rate": 1.9149618543175083e-05,
49
+ "loss": 0.7798,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.19,
54
+ "learning_rate": 1.902813547791438e-05,
55
+ "loss": 0.8225,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.22,
60
+ "learning_rate": 1.8906652412653678e-05,
61
+ "loss": 0.7332,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 0.24,
66
+ "learning_rate": 1.8785169347392975e-05,
67
+ "loss": 0.7285,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 0.27,
72
+ "learning_rate": 1.8663686282132272e-05,
73
+ "loss": 0.7375,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 0.29,
78
+ "learning_rate": 1.854220321687157e-05,
79
+ "loss": 0.7397,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 0.32,
84
+ "learning_rate": 1.8420720151610867e-05,
85
+ "loss": 0.709,
86
+ "step": 6500
87
+ },
88
+ {
89
+ "epoch": 0.34,
90
+ "learning_rate": 1.8299237086350164e-05,
91
+ "loss": 0.7459,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 0.36,
96
+ "learning_rate": 1.8177754021089462e-05,
97
+ "loss": 0.727,
98
+ "step": 7500
99
+ },
100
+ {
101
+ "epoch": 0.39,
102
+ "learning_rate": 1.805627095582876e-05,
103
+ "loss": 0.7154,
104
+ "step": 8000
105
+ },
106
+ {
107
+ "epoch": 0.41,
108
+ "learning_rate": 1.7934787890568056e-05,
109
+ "loss": 0.7378,
110
+ "step": 8500
111
+ },
112
+ {
113
+ "epoch": 0.44,
114
+ "learning_rate": 1.7813304825307354e-05,
115
+ "loss": 0.727,
116
+ "step": 9000
117
+ },
118
+ {
119
+ "epoch": 0.46,
120
+ "learning_rate": 1.769182176004665e-05,
121
+ "loss": 0.6885,
122
+ "step": 9500
123
+ },
124
+ {
125
+ "epoch": 0.49,
126
+ "learning_rate": 1.757033869478595e-05,
127
+ "loss": 0.6895,
128
+ "step": 10000
129
+ },
130
+ {
131
+ "epoch": 0.51,
132
+ "learning_rate": 1.7448855629525246e-05,
133
+ "loss": 0.6916,
134
+ "step": 10500
135
+ },
136
+ {
137
+ "epoch": 0.53,
138
+ "learning_rate": 1.7327372564264543e-05,
139
+ "loss": 0.7228,
140
+ "step": 11000
141
+ },
142
+ {
143
+ "epoch": 0.56,
144
+ "learning_rate": 1.720588949900384e-05,
145
+ "loss": 0.6772,
146
+ "step": 11500
147
+ },
148
+ {
149
+ "epoch": 0.58,
150
+ "learning_rate": 1.7084406433743138e-05,
151
+ "loss": 0.6988,
152
+ "step": 12000
153
+ },
154
+ {
155
+ "epoch": 0.61,
156
+ "learning_rate": 1.6962923368482435e-05,
157
+ "loss": 0.7078,
158
+ "step": 12500
159
+ },
160
+ {
161
+ "epoch": 0.63,
162
+ "learning_rate": 1.6841440303221732e-05,
163
+ "loss": 0.726,
164
+ "step": 13000
165
+ },
166
+ {
167
+ "epoch": 0.66,
168
+ "learning_rate": 1.671995723796103e-05,
169
+ "loss": 0.7046,
170
+ "step": 13500
171
+ },
172
+ {
173
+ "epoch": 0.68,
174
+ "learning_rate": 1.6598474172700327e-05,
175
+ "loss": 0.7028,
176
+ "step": 14000
177
+ },
178
+ {
179
+ "epoch": 0.7,
180
+ "learning_rate": 1.6476991107439624e-05,
181
+ "loss": 0.6893,
182
+ "step": 14500
183
+ },
184
+ {
185
+ "epoch": 0.73,
186
+ "learning_rate": 1.6355508042178922e-05,
187
+ "loss": 0.7066,
188
+ "step": 15000
189
+ },
190
+ {
191
+ "epoch": 0.75,
192
+ "learning_rate": 1.623402497691822e-05,
193
+ "loss": 0.7021,
194
+ "step": 15500
195
+ },
196
+ {
197
+ "epoch": 0.78,
198
+ "learning_rate": 1.6112541911657517e-05,
199
+ "loss": 0.718,
200
+ "step": 16000
201
+ },
202
+ {
203
+ "epoch": 0.8,
204
+ "learning_rate": 1.5991058846396814e-05,
205
+ "loss": 0.6963,
206
+ "step": 16500
207
+ },
208
+ {
209
+ "epoch": 0.83,
210
+ "learning_rate": 1.586957578113611e-05,
211
+ "loss": 0.664,
212
+ "step": 17000
213
+ },
214
+ {
215
+ "epoch": 0.85,
216
+ "learning_rate": 1.574809271587541e-05,
217
+ "loss": 0.6841,
218
+ "step": 17500
219
+ },
220
+ {
221
+ "epoch": 0.87,
222
+ "learning_rate": 1.5626609650614706e-05,
223
+ "loss": 0.7121,
224
+ "step": 18000
225
+ },
226
+ {
227
+ "epoch": 0.9,
228
+ "learning_rate": 1.5505126585354003e-05,
229
+ "loss": 0.6782,
230
+ "step": 18500
231
+ },
232
+ {
233
+ "epoch": 0.92,
234
+ "learning_rate": 1.53836435200933e-05,
235
+ "loss": 0.7029,
236
+ "step": 19000
237
+ },
238
+ {
239
+ "epoch": 0.95,
240
+ "learning_rate": 1.5262160454832598e-05,
241
+ "loss": 0.7159,
242
+ "step": 19500
243
+ },
244
+ {
245
+ "epoch": 0.97,
246
+ "learning_rate": 1.5140677389571894e-05,
247
+ "loss": 0.6624,
248
+ "step": 20000
249
+ },
250
+ {
251
+ "epoch": 1.0,
252
+ "learning_rate": 1.5019194324311193e-05,
253
+ "loss": 0.6956,
254
+ "step": 20500
255
+ },
256
+ {
257
+ "epoch": 1.02,
258
+ "learning_rate": 1.489771125905049e-05,
259
+ "loss": 0.6183,
260
+ "step": 21000
261
+ },
262
+ {
263
+ "epoch": 1.04,
264
+ "learning_rate": 1.4776228193789787e-05,
265
+ "loss": 0.6394,
266
+ "step": 21500
267
+ },
268
+ {
269
+ "epoch": 1.07,
270
+ "learning_rate": 1.4654745128529085e-05,
271
+ "loss": 0.6252,
272
+ "step": 22000
273
+ },
274
+ {
275
+ "epoch": 1.09,
276
+ "learning_rate": 1.453326206326838e-05,
277
+ "loss": 0.623,
278
+ "step": 22500
279
+ },
280
+ {
281
+ "epoch": 1.12,
282
+ "learning_rate": 1.441177899800768e-05,
283
+ "loss": 0.6187,
284
+ "step": 23000
285
+ },
286
+ {
287
+ "epoch": 1.14,
288
+ "learning_rate": 1.4290295932746977e-05,
289
+ "loss": 0.5981,
290
+ "step": 23500
291
+ },
292
+ {
293
+ "epoch": 1.17,
294
+ "learning_rate": 1.4168812867486274e-05,
295
+ "loss": 0.6317,
296
+ "step": 24000
297
+ },
298
+ {
299
+ "epoch": 1.19,
300
+ "learning_rate": 1.4047329802225571e-05,
301
+ "loss": 0.6205,
302
+ "step": 24500
303
+ },
304
+ {
305
+ "epoch": 1.21,
306
+ "learning_rate": 1.3925846736964867e-05,
307
+ "loss": 0.5987,
308
+ "step": 25000
309
+ },
310
+ {
311
+ "epoch": 1.24,
312
+ "learning_rate": 1.3804363671704166e-05,
313
+ "loss": 0.6027,
314
+ "step": 25500
315
+ },
316
+ {
317
+ "epoch": 1.26,
318
+ "learning_rate": 1.3682880606443463e-05,
319
+ "loss": 0.6122,
320
+ "step": 26000
321
+ },
322
+ {
323
+ "epoch": 1.29,
324
+ "learning_rate": 1.3561397541182759e-05,
325
+ "loss": 0.6165,
326
+ "step": 26500
327
+ },
328
+ {
329
+ "epoch": 1.31,
330
+ "learning_rate": 1.3439914475922058e-05,
331
+ "loss": 0.6176,
332
+ "step": 27000
333
+ },
334
+ {
335
+ "epoch": 1.34,
336
+ "learning_rate": 1.3318431410661354e-05,
337
+ "loss": 0.6036,
338
+ "step": 27500
339
+ },
340
+ {
341
+ "epoch": 1.36,
342
+ "learning_rate": 1.3196948345400653e-05,
343
+ "loss": 0.6088,
344
+ "step": 28000
345
+ },
346
+ {
347
+ "epoch": 1.38,
348
+ "learning_rate": 1.307546528013995e-05,
349
+ "loss": 0.6363,
350
+ "step": 28500
351
+ },
352
+ {
353
+ "epoch": 1.41,
354
+ "learning_rate": 1.2953982214879246e-05,
355
+ "loss": 0.603,
356
+ "step": 29000
357
+ },
358
+ {
359
+ "epoch": 1.43,
360
+ "learning_rate": 1.2832499149618545e-05,
361
+ "loss": 0.658,
362
+ "step": 29500
363
+ },
364
+ {
365
+ "epoch": 1.46,
366
+ "learning_rate": 1.271101608435784e-05,
367
+ "loss": 0.6219,
368
+ "step": 30000
369
+ },
370
+ {
371
+ "epoch": 1.48,
372
+ "learning_rate": 1.258953301909714e-05,
373
+ "loss": 0.5905,
374
+ "step": 30500
375
+ },
376
+ {
377
+ "epoch": 1.51,
378
+ "learning_rate": 1.2468049953836437e-05,
379
+ "loss": 0.6223,
380
+ "step": 31000
381
+ },
382
+ {
383
+ "epoch": 1.53,
384
+ "learning_rate": 1.2346566888575732e-05,
385
+ "loss": 0.6117,
386
+ "step": 31500
387
+ },
388
+ {
389
+ "epoch": 1.55,
390
+ "learning_rate": 1.2225083823315031e-05,
391
+ "loss": 0.5922,
392
+ "step": 32000
393
+ },
394
+ {
395
+ "epoch": 1.58,
396
+ "learning_rate": 1.2103600758054329e-05,
397
+ "loss": 0.5848,
398
+ "step": 32500
399
+ },
400
+ {
401
+ "epoch": 1.6,
402
+ "learning_rate": 1.1982117692793624e-05,
403
+ "loss": 0.5924,
404
+ "step": 33000
405
+ },
406
+ {
407
+ "epoch": 1.63,
408
+ "learning_rate": 1.1860634627532923e-05,
409
+ "loss": 0.5936,
410
+ "step": 33500
411
+ },
412
+ {
413
+ "epoch": 1.65,
414
+ "learning_rate": 1.1739151562272219e-05,
415
+ "loss": 0.6374,
416
+ "step": 34000
417
+ },
418
+ {
419
+ "epoch": 1.68,
420
+ "learning_rate": 1.1617668497011518e-05,
421
+ "loss": 0.5841,
422
+ "step": 34500
423
+ },
424
+ {
425
+ "epoch": 1.7,
426
+ "learning_rate": 1.1496185431750815e-05,
427
+ "loss": 0.5954,
428
+ "step": 35000
429
+ },
430
+ {
431
+ "epoch": 1.73,
432
+ "learning_rate": 1.1374702366490111e-05,
433
+ "loss": 0.6168,
434
+ "step": 35500
435
+ },
436
+ {
437
+ "epoch": 1.75,
438
+ "learning_rate": 1.125321930122941e-05,
439
+ "loss": 0.6063,
440
+ "step": 36000
441
+ },
442
+ {
443
+ "epoch": 1.77,
444
+ "learning_rate": 1.1131736235968706e-05,
445
+ "loss": 0.5914,
446
+ "step": 36500
447
+ },
448
+ {
449
+ "epoch": 1.8,
450
+ "learning_rate": 1.1010253170708005e-05,
451
+ "loss": 0.6424,
452
+ "step": 37000
453
+ },
454
+ {
455
+ "epoch": 1.82,
456
+ "learning_rate": 1.0888770105447302e-05,
457
+ "loss": 0.6359,
458
+ "step": 37500
459
+ },
460
+ {
461
+ "epoch": 1.85,
462
+ "learning_rate": 1.0767287040186598e-05,
463
+ "loss": 0.6023,
464
+ "step": 38000
465
+ },
466
+ {
467
+ "epoch": 1.87,
468
+ "learning_rate": 1.0645803974925897e-05,
469
+ "loss": 0.5855,
470
+ "step": 38500
471
+ },
472
+ {
473
+ "epoch": 1.9,
474
+ "learning_rate": 1.0524320909665192e-05,
475
+ "loss": 0.6153,
476
+ "step": 39000
477
+ },
478
+ {
479
+ "epoch": 1.92,
480
+ "learning_rate": 1.0402837844404491e-05,
481
+ "loss": 0.6025,
482
+ "step": 39500
483
+ },
484
+ {
485
+ "epoch": 1.94,
486
+ "learning_rate": 1.0281354779143789e-05,
487
+ "loss": 0.6214,
488
+ "step": 40000
489
+ },
490
+ {
491
+ "epoch": 1.97,
492
+ "learning_rate": 1.0159871713883084e-05,
493
+ "loss": 0.6067,
494
+ "step": 40500
495
+ },
496
+ {
497
+ "epoch": 1.99,
498
+ "learning_rate": 1.0038388648622384e-05,
499
+ "loss": 0.6057,
500
+ "step": 41000
501
+ },
502
+ {
503
+ "epoch": 2.02,
504
+ "learning_rate": 9.916905583361681e-06,
505
+ "loss": 0.5468,
506
+ "step": 41500
507
+ },
508
+ {
509
+ "epoch": 2.04,
510
+ "learning_rate": 9.795422518100978e-06,
511
+ "loss": 0.4872,
512
+ "step": 42000
513
+ },
514
+ {
515
+ "epoch": 2.07,
516
+ "learning_rate": 9.673939452840276e-06,
517
+ "loss": 0.5144,
518
+ "step": 42500
519
+ },
520
+ {
521
+ "epoch": 2.09,
522
+ "learning_rate": 9.552456387579571e-06,
523
+ "loss": 0.5122,
524
+ "step": 43000
525
+ },
526
+ {
527
+ "epoch": 2.11,
528
+ "learning_rate": 9.430973322318869e-06,
529
+ "loss": 0.5152,
530
+ "step": 43500
531
+ },
532
+ {
533
+ "epoch": 2.14,
534
+ "learning_rate": 9.309490257058168e-06,
535
+ "loss": 0.5245,
536
+ "step": 44000
537
+ },
538
+ {
539
+ "epoch": 2.16,
540
+ "learning_rate": 9.188007191797465e-06,
541
+ "loss": 0.4971,
542
+ "step": 44500
543
+ },
544
+ {
545
+ "epoch": 2.19,
546
+ "learning_rate": 9.066524126536762e-06,
547
+ "loss": 0.5478,
548
+ "step": 45000
549
+ },
550
+ {
551
+ "epoch": 2.21,
552
+ "learning_rate": 8.945041061276058e-06,
553
+ "loss": 0.5296,
554
+ "step": 45500
555
+ },
556
+ {
557
+ "epoch": 2.24,
558
+ "learning_rate": 8.823557996015357e-06,
559
+ "loss": 0.5403,
560
+ "step": 46000
561
+ },
562
+ {
563
+ "epoch": 2.26,
564
+ "learning_rate": 8.702074930754654e-06,
565
+ "loss": 0.5133,
566
+ "step": 46500
567
+ },
568
+ {
569
+ "epoch": 2.28,
570
+ "learning_rate": 8.580591865493952e-06,
571
+ "loss": 0.5162,
572
+ "step": 47000
573
+ },
574
+ {
575
+ "epoch": 2.31,
576
+ "learning_rate": 8.459108800233247e-06,
577
+ "loss": 0.521,
578
+ "step": 47500
579
+ },
580
+ {
581
+ "epoch": 2.33,
582
+ "learning_rate": 8.337625734972545e-06,
583
+ "loss": 0.4896,
584
+ "step": 48000
585
+ },
586
+ {
587
+ "epoch": 2.36,
588
+ "learning_rate": 8.216142669711844e-06,
589
+ "loss": 0.528,
590
+ "step": 48500
591
+ },
592
+ {
593
+ "epoch": 2.38,
594
+ "learning_rate": 8.094659604451141e-06,
595
+ "loss": 0.5116,
596
+ "step": 49000
597
+ },
598
+ {
599
+ "epoch": 2.41,
600
+ "learning_rate": 7.973176539190437e-06,
601
+ "loss": 0.5467,
602
+ "step": 49500
603
+ },
604
+ {
605
+ "epoch": 2.43,
606
+ "learning_rate": 7.851693473929734e-06,
607
+ "loss": 0.5073,
608
+ "step": 50000
609
+ },
610
+ {
611
+ "epoch": 2.45,
612
+ "learning_rate": 7.730210408669033e-06,
613
+ "loss": 0.544,
614
+ "step": 50500
615
+ },
616
+ {
617
+ "epoch": 2.48,
618
+ "learning_rate": 7.6087273434083295e-06,
619
+ "loss": 0.5209,
620
+ "step": 51000
621
+ },
622
+ {
623
+ "epoch": 2.5,
624
+ "learning_rate": 7.487244278147628e-06,
625
+ "loss": 0.514,
626
+ "step": 51500
627
+ },
628
+ {
629
+ "epoch": 2.53,
630
+ "learning_rate": 7.365761212886924e-06,
631
+ "loss": 0.5189,
632
+ "step": 52000
633
+ },
634
+ {
635
+ "epoch": 2.55,
636
+ "learning_rate": 7.2442781476262215e-06,
637
+ "loss": 0.5159,
638
+ "step": 52500
639
+ },
640
+ {
641
+ "epoch": 2.58,
642
+ "learning_rate": 7.122795082365519e-06,
643
+ "loss": 0.4995,
644
+ "step": 53000
645
+ },
646
+ {
647
+ "epoch": 2.6,
648
+ "learning_rate": 7.001312017104816e-06,
649
+ "loss": 0.5226,
650
+ "step": 53500
651
+ },
652
+ {
653
+ "epoch": 2.62,
654
+ "learning_rate": 6.879828951844113e-06,
655
+ "loss": 0.4936,
656
+ "step": 54000
657
+ },
658
+ {
659
+ "epoch": 2.65,
660
+ "learning_rate": 6.758345886583411e-06,
661
+ "loss": 0.5277,
662
+ "step": 54500
663
+ },
664
+ {
665
+ "epoch": 2.67,
666
+ "learning_rate": 6.636862821322708e-06,
667
+ "loss": 0.5007,
668
+ "step": 55000
669
+ },
670
+ {
671
+ "epoch": 2.7,
672
+ "learning_rate": 6.5153797560620055e-06,
673
+ "loss": 0.5108,
674
+ "step": 55500
675
+ },
676
+ {
677
+ "epoch": 2.72,
678
+ "learning_rate": 6.393896690801304e-06,
679
+ "loss": 0.5149,
680
+ "step": 56000
681
+ },
682
+ {
683
+ "epoch": 2.75,
684
+ "learning_rate": 6.2724136255406e-06,
685
+ "loss": 0.5111,
686
+ "step": 56500
687
+ },
688
+ {
689
+ "epoch": 2.77,
690
+ "learning_rate": 6.1509305602798975e-06,
691
+ "loss": 0.5132,
692
+ "step": 57000
693
+ },
694
+ {
695
+ "epoch": 2.79,
696
+ "learning_rate": 6.029447495019195e-06,
697
+ "loss": 0.5067,
698
+ "step": 57500
699
+ },
700
+ {
701
+ "epoch": 2.82,
702
+ "learning_rate": 5.907964429758492e-06,
703
+ "loss": 0.5027,
704
+ "step": 58000
705
+ },
706
+ {
707
+ "epoch": 2.84,
708
+ "learning_rate": 5.786481364497789e-06,
709
+ "loss": 0.5302,
710
+ "step": 58500
711
+ },
712
+ {
713
+ "epoch": 2.87,
714
+ "learning_rate": 5.664998299237087e-06,
715
+ "loss": 0.477,
716
+ "step": 59000
717
+ },
718
+ {
719
+ "epoch": 2.89,
720
+ "learning_rate": 5.543515233976384e-06,
721
+ "loss": 0.514,
722
+ "step": 59500
723
+ },
724
+ {
725
+ "epoch": 2.92,
726
+ "learning_rate": 5.422032168715682e-06,
727
+ "loss": 0.5108,
728
+ "step": 60000
729
+ },
730
+ {
731
+ "epoch": 2.94,
732
+ "learning_rate": 5.300549103454978e-06,
733
+ "loss": 0.52,
734
+ "step": 60500
735
+ },
736
+ {
737
+ "epoch": 2.96,
738
+ "learning_rate": 5.179066038194276e-06,
739
+ "loss": 0.5177,
740
+ "step": 61000
741
+ },
742
+ {
743
+ "epoch": 2.99,
744
+ "learning_rate": 5.057582972933574e-06,
745
+ "loss": 0.5323,
746
+ "step": 61500
747
+ },
748
+ {
749
+ "epoch": 3.01,
750
+ "learning_rate": 4.936099907672871e-06,
751
+ "loss": 0.4662,
752
+ "step": 62000
753
+ },
754
+ {
755
+ "epoch": 3.04,
756
+ "learning_rate": 4.814616842412168e-06,
757
+ "loss": 0.4248,
758
+ "step": 62500
759
+ },
760
+ {
761
+ "epoch": 3.06,
762
+ "learning_rate": 4.693133777151466e-06,
763
+ "loss": 0.3953,
764
+ "step": 63000
765
+ },
766
+ {
767
+ "epoch": 3.09,
768
+ "learning_rate": 4.571650711890763e-06,
769
+ "loss": 0.4326,
770
+ "step": 63500
771
+ },
772
+ {
773
+ "epoch": 3.11,
774
+ "learning_rate": 4.45016764663006e-06,
775
+ "loss": 0.4445,
776
+ "step": 64000
777
+ },
778
+ {
779
+ "epoch": 3.13,
780
+ "learning_rate": 4.328684581369358e-06,
781
+ "loss": 0.4099,
782
+ "step": 64500
783
+ },
784
+ {
785
+ "epoch": 3.16,
786
+ "learning_rate": 4.207201516108655e-06,
787
+ "loss": 0.433,
788
+ "step": 65000
789
+ },
790
+ {
791
+ "epoch": 3.18,
792
+ "learning_rate": 4.085718450847952e-06,
793
+ "loss": 0.4228,
794
+ "step": 65500
795
+ },
796
+ {
797
+ "epoch": 3.21,
798
+ "learning_rate": 3.96423538558725e-06,
799
+ "loss": 0.4197,
800
+ "step": 66000
801
+ },
802
+ {
803
+ "epoch": 3.23,
804
+ "learning_rate": 3.842752320326546e-06,
805
+ "loss": 0.4482,
806
+ "step": 66500
807
+ },
808
+ {
809
+ "epoch": 3.26,
810
+ "learning_rate": 3.7212692550658443e-06,
811
+ "loss": 0.4415,
812
+ "step": 67000
813
+ },
814
+ {
815
+ "epoch": 3.28,
816
+ "learning_rate": 3.5997861898051417e-06,
817
+ "loss": 0.4289,
818
+ "step": 67500
819
+ },
820
+ {
821
+ "epoch": 3.3,
822
+ "learning_rate": 3.4783031245444386e-06,
823
+ "loss": 0.4446,
824
+ "step": 68000
825
+ },
826
+ {
827
+ "epoch": 3.33,
828
+ "learning_rate": 3.3568200592837364e-06,
829
+ "loss": 0.4625,
830
+ "step": 68500
831
+ },
832
+ {
833
+ "epoch": 3.35,
834
+ "learning_rate": 3.2353369940230333e-06,
835
+ "loss": 0.3993,
836
+ "step": 69000
837
+ },
838
+ {
839
+ "epoch": 3.38,
840
+ "learning_rate": 3.113853928762331e-06,
841
+ "loss": 0.4154,
842
+ "step": 69500
843
+ },
844
+ {
845
+ "epoch": 3.4,
846
+ "learning_rate": 2.992370863501628e-06,
847
+ "loss": 0.4245,
848
+ "step": 70000
849
+ },
850
+ {
851
+ "epoch": 3.43,
852
+ "learning_rate": 2.8708877982409257e-06,
853
+ "loss": 0.4084,
854
+ "step": 70500
855
+ },
856
+ {
857
+ "epoch": 3.45,
858
+ "learning_rate": 2.7494047329802226e-06,
859
+ "loss": 0.4191,
860
+ "step": 71000
861
+ },
862
+ {
863
+ "epoch": 3.47,
864
+ "learning_rate": 2.62792166771952e-06,
865
+ "loss": 0.4498,
866
+ "step": 71500
867
+ },
868
+ {
869
+ "epoch": 3.5,
870
+ "learning_rate": 2.5064386024588173e-06,
871
+ "loss": 0.4296,
872
+ "step": 72000
873
+ },
874
+ {
875
+ "epoch": 3.52,
876
+ "learning_rate": 2.3849555371981147e-06,
877
+ "loss": 0.456,
878
+ "step": 72500
879
+ },
880
+ {
881
+ "epoch": 3.55,
882
+ "learning_rate": 2.263472471937412e-06,
883
+ "loss": 0.423,
884
+ "step": 73000
885
+ },
886
+ {
887
+ "epoch": 3.57,
888
+ "learning_rate": 2.1419894066767093e-06,
889
+ "loss": 0.4062,
890
+ "step": 73500
891
+ },
892
+ {
893
+ "epoch": 3.6,
894
+ "learning_rate": 2.0205063414160067e-06,
895
+ "loss": 0.435,
896
+ "step": 74000
897
+ },
898
+ {
899
+ "epoch": 3.62,
900
+ "learning_rate": 1.899023276155304e-06,
901
+ "loss": 0.4221,
902
+ "step": 74500
903
+ },
904
+ {
905
+ "epoch": 3.64,
906
+ "learning_rate": 1.7775402108946014e-06,
907
+ "loss": 0.4444,
908
+ "step": 75000
909
+ },
910
+ {
911
+ "epoch": 3.67,
912
+ "learning_rate": 1.6560571456338987e-06,
913
+ "loss": 0.4031,
914
+ "step": 75500
915
+ },
916
+ {
917
+ "epoch": 3.69,
918
+ "learning_rate": 1.5345740803731962e-06,
919
+ "loss": 0.4238,
920
+ "step": 76000
921
+ },
922
+ {
923
+ "epoch": 3.72,
924
+ "learning_rate": 1.4130910151124936e-06,
925
+ "loss": 0.4356,
926
+ "step": 76500
927
+ },
928
+ {
929
+ "epoch": 3.74,
930
+ "learning_rate": 1.291607949851791e-06,
931
+ "loss": 0.4085,
932
+ "step": 77000
933
+ },
934
+ {
935
+ "epoch": 3.77,
936
+ "learning_rate": 1.170124884591088e-06,
937
+ "loss": 0.4056,
938
+ "step": 77500
939
+ },
940
+ {
941
+ "epoch": 3.79,
942
+ "learning_rate": 1.0486418193303854e-06,
943
+ "loss": 0.432,
944
+ "step": 78000
945
+ },
946
+ {
947
+ "epoch": 3.81,
948
+ "learning_rate": 9.271587540696827e-07,
949
+ "loss": 0.4142,
950
+ "step": 78500
951
+ },
952
+ {
953
+ "epoch": 3.84,
954
+ "learning_rate": 8.056756888089801e-07,
955
+ "loss": 0.4236,
956
+ "step": 79000
957
+ },
958
+ {
959
+ "epoch": 3.86,
960
+ "learning_rate": 6.841926235482774e-07,
961
+ "loss": 0.4183,
962
+ "step": 79500
963
+ },
964
+ {
965
+ "epoch": 3.89,
966
+ "learning_rate": 5.627095582875748e-07,
967
+ "loss": 0.4129,
968
+ "step": 80000
969
+ },
970
+ {
971
+ "epoch": 3.91,
972
+ "learning_rate": 4.4122649302687214e-07,
973
+ "loss": 0.4052,
974
+ "step": 80500
975
+ },
976
+ {
977
+ "epoch": 3.94,
978
+ "learning_rate": 3.1974342776616943e-07,
979
+ "loss": 0.4121,
980
+ "step": 81000
981
+ },
982
+ {
983
+ "epoch": 3.96,
984
+ "learning_rate": 1.9826036250546677e-07,
985
+ "loss": 0.3846,
986
+ "step": 81500
987
+ },
988
+ {
989
+ "epoch": 3.98,
990
+ "learning_rate": 7.677729724476409e-08,
991
+ "loss": 0.4475,
992
+ "step": 82000
993
+ },
994
+ {
995
+ "epoch": 4.0,
996
+ "step": 82316,
997
+ "total_flos": 1.2905751724355635e+17,
998
+ "train_loss": 0.5708848007960676,
999
+ "train_runtime": 17933.1446,
1000
+ "train_samples_per_second": 36.722,
1001
+ "train_steps_per_second": 4.59
1002
+ }
1003
+ ],
1004
+ "max_steps": 82316,
1005
+ "num_train_epochs": 4,
1006
+ "total_flos": 1.2905751724355635e+17,
1007
+ "trial_name": null,
1008
+ "trial_params": null
1009
+ }
deberta_qa/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:992fc0b718af1b638e2d7e8dc932971da2af84aa1983a79087faf46cb17c0c75
3
+ size 3579
distilbart_summarizer/README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: distilbart-cnn-12-6_1e-5
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # distilbart-cnn-12-6_1e-5
16
+
17
+ This model is a fine-tuned version of [sshleifer/distilbart-cnn-12-6](https://huggingface.co/sshleifer/distilbart-cnn-12-6) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.0778
20
+ - Rouge1: 51.4305
21
+ - Rouge2: 30.8023
22
+ - Rougel: 40.977
23
+ - Rougelsum: 45.5778
24
+ - Gen Len: 79.9607
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - learning_rate: 1e-05
44
+ - train_batch_size: 4
45
+ - eval_batch_size: 4
46
+ - seed: 42
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: linear
49
+ - num_epochs: 3.0
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
54
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
55
+ | 1.2906 | 0.62 | 500 | 1.1144 | 49.7349 | 29.2413 | 39.062 | 43.9746 | 81.3764 |
56
+ | 1.0677 | 1.25 | 1000 | 1.0895 | 50.4239 | 30.3398 | 40.2029 | 44.8709 | 80.7528 |
57
+ | 0.9725 | 1.87 | 1500 | 1.0740 | 51.3474 | 31.1253 | 40.79 | 45.717 | 81.8455 |
58
+ | 0.8629 | 2.5 | 2000 | 1.0816 | 50.8972 | 30.6779 | 40.4662 | 45.3636 | 81.7191 |
59
+
60
+
61
+ ### Framework versions
62
+
63
+ - Transformers 4.27.0.dev0
64
+ - Pytorch 1.13.1+cu117
65
+ - Datasets 2.10.0
66
+ - Tokenizers 0.13.2
distilbart_summarizer/all_results.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_gen_len": 79.96067415730337,
4
+ "eval_loss": 1.0778131484985352,
5
+ "eval_rouge1": 51.4305,
6
+ "eval_rouge2": 30.8023,
7
+ "eval_rougeL": 40.977,
8
+ "eval_rougeLsum": 45.5778,
9
+ "eval_runtime": 172.4056,
10
+ "eval_samples": 356,
11
+ "eval_samples_per_second": 2.065,
12
+ "eval_steps_per_second": 0.516,
13
+ "train_loss": 1.0117533442084115,
14
+ "train_runtime": 1656.9583,
15
+ "train_samples": 3204,
16
+ "train_samples_per_second": 5.801,
17
+ "train_steps_per_second": 1.45
18
+ }
distilbart_summarizer/config.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "sshleifer/distilbart-cnn-12-6",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": false,
8
+ "architectures": [
9
+ "BartForConditionalGeneration"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bos_token_id": 0,
13
+ "classif_dropout": 0.0,
14
+ "classifier_dropout": 0.0,
15
+ "d_model": 1024,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 6,
20
+ "decoder_start_token_id": 2,
21
+ "dropout": 0.1,
22
+ "early_stopping": true,
23
+ "encoder_attention_heads": 16,
24
+ "encoder_ffn_dim": 4096,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 12,
27
+ "eos_token_id": 2,
28
+ "extra_pos_embeddings": 2,
29
+ "force_bos_token_to_be_generated": true,
30
+ "forced_bos_token_id": 0,
31
+ "forced_eos_token_id": 2,
32
+ "gradient_checkpointing": false,
33
+ "id2label": {
34
+ "0": "LABEL_0",
35
+ "1": "LABEL_1",
36
+ "2": "LABEL_2"
37
+ },
38
+ "init_std": 0.02,
39
+ "is_encoder_decoder": true,
40
+ "label2id": {
41
+ "LABEL_0": 0,
42
+ "LABEL_1": 1,
43
+ "LABEL_2": 2
44
+ },
45
+ "length_penalty": 2.0,
46
+ "max_length": 142,
47
+ "max_position_embeddings": 1024,
48
+ "min_length": 56,
49
+ "model_type": "bart",
50
+ "no_repeat_ngram_size": 3,
51
+ "normalize_before": false,
52
+ "normalize_embedding": true,
53
+ "num_beams": 4,
54
+ "num_hidden_layers": 12,
55
+ "output_past": true,
56
+ "pad_token_id": 1,
57
+ "prefix": " ",
58
+ "replacing_rate": 0,
59
+ "scale_embedding": false,
60
+ "static_position_embeddings": false,
61
+ "student_decoder_layers": null,
62
+ "student_encoder_layers": null,
63
+ "task_specific_params": {
64
+ "summarization": {
65
+ "early_stopping": true,
66
+ "length_penalty": 2.0,
67
+ "max_length": 142,
68
+ "min_length": 56,
69
+ "no_repeat_ngram_size": 3,
70
+ "num_beams": 4
71
+ }
72
+ },
73
+ "torch_dtype": "float32",
74
+ "transformers_version": "4.27.0.dev0",
75
+ "use_cache": true,
76
+ "vocab_size": 50265
77
+ }
distilbart_summarizer/eval_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_gen_len": 79.96067415730337,
4
+ "eval_loss": 1.0778131484985352,
5
+ "eval_rouge1": 51.4305,
6
+ "eval_rouge2": 30.8023,
7
+ "eval_rougeL": 40.977,
8
+ "eval_rougeLsum": 45.5778,
9
+ "eval_runtime": 172.4056,
10
+ "eval_samples": 356,
11
+ "eval_samples_per_second": 2.065,
12
+ "eval_steps_per_second": 0.516
13
+ }
distilbart_summarizer/generation_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 2,
4
+ "early_stopping": true,
5
+ "eos_token_id": 2,
6
+ "forced_bos_token_id": 0,
7
+ "forced_eos_token_id": 2,
8
+ "length_penalty": 2.0,
9
+ "max_length": 142,
10
+ "min_length": 56,
11
+ "no_repeat_ngram_size": 3,
12
+ "num_beams": 4,
13
+ "pad_token_id": 1,
14
+ "transformers_version": "4.27.0.dev0"
15
+ }
distilbart_summarizer/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
distilbart_summarizer/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33aff931672163e076cbe77981c2b52b770a4b1047f0df7b2203838a9fa59ded
3
+ size 1222367837
distilbart_summarizer/special_tokens_map.json ADDED
@@ -0,0 +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
+ }
distilbart_summarizer/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
distilbart_summarizer/tokenizer_config.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": {
4
+ "__type": "AddedToken",
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false
10
+ },
11
+ "cls_token": {
12
+ "__type": "AddedToken",
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "eos_token": {
20
+ "__type": "AddedToken",
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ },
27
+ "errors": "replace",
28
+ "mask_token": {
29
+ "__type": "AddedToken",
30
+ "content": "<mask>",
31
+ "lstrip": true,
32
+ "normalized": true,
33
+ "rstrip": false,
34
+ "single_word": false
35
+ },
36
+ "model_max_length": 1024,
37
+ "pad_token": {
38
+ "__type": "AddedToken",
39
+ "content": "<pad>",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false
44
+ },
45
+ "sep_token": {
46
+ "__type": "AddedToken",
47
+ "content": "</s>",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false
52
+ },
53
+ "special_tokens_map_file": null,
54
+ "tokenizer_class": "BartTokenizer",
55
+ "trim_offsets": true,
56
+ "unk_token": {
57
+ "__type": "AddedToken",
58
+ "content": "<unk>",
59
+ "lstrip": false,
60
+ "normalized": true,
61
+ "rstrip": false,
62
+ "single_word": false
63
+ }
64
+ }
distilbart_summarizer/train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "train_loss": 1.0117533442084115,
4
+ "train_runtime": 1656.9583,
5
+ "train_samples": 3204,
6
+ "train_samples_per_second": 5.801,
7
+ "train_steps_per_second": 1.45
8
+ }
distilbart_summarizer/trainer_state.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "global_step": 2403,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.62,
12
+ "learning_rate": 7.919267582188932e-06,
13
+ "loss": 1.2906,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.62,
18
+ "eval_gen_len": 81.37640449438203,
19
+ "eval_loss": 1.1144022941589355,
20
+ "eval_rouge1": 49.7349,
21
+ "eval_rouge2": 29.2413,
22
+ "eval_rougeL": 39.062,
23
+ "eval_rougeLsum": 43.9746,
24
+ "eval_runtime": 171.8864,
25
+ "eval_samples_per_second": 2.071,
26
+ "eval_steps_per_second": 0.518,
27
+ "step": 500
28
+ },
29
+ {
30
+ "epoch": 1.25,
31
+ "learning_rate": 5.838535164377861e-06,
32
+ "loss": 1.0677,
33
+ "step": 1000
34
+ },
35
+ {
36
+ "epoch": 1.25,
37
+ "eval_gen_len": 80.75280898876404,
38
+ "eval_loss": 1.0894856452941895,
39
+ "eval_rouge1": 50.4239,
40
+ "eval_rouge2": 30.3398,
41
+ "eval_rougeL": 40.2029,
42
+ "eval_rougeLsum": 44.8709,
43
+ "eval_runtime": 175.8209,
44
+ "eval_samples_per_second": 2.025,
45
+ "eval_steps_per_second": 0.506,
46
+ "step": 1000
47
+ },
48
+ {
49
+ "epoch": 1.87,
50
+ "learning_rate": 3.7578027465667915e-06,
51
+ "loss": 0.9725,
52
+ "step": 1500
53
+ },
54
+ {
55
+ "epoch": 1.87,
56
+ "eval_gen_len": 81.84550561797752,
57
+ "eval_loss": 1.0739564895629883,
58
+ "eval_rouge1": 51.3474,
59
+ "eval_rouge2": 31.1253,
60
+ "eval_rougeL": 40.79,
61
+ "eval_rougeLsum": 45.717,
62
+ "eval_runtime": 173.2132,
63
+ "eval_samples_per_second": 2.055,
64
+ "eval_steps_per_second": 0.514,
65
+ "step": 1500
66
+ },
67
+ {
68
+ "epoch": 2.5,
69
+ "learning_rate": 1.677070328755722e-06,
70
+ "loss": 0.8629,
71
+ "step": 2000
72
+ },
73
+ {
74
+ "epoch": 2.5,
75
+ "eval_gen_len": 81.71910112359551,
76
+ "eval_loss": 1.0816434621810913,
77
+ "eval_rouge1": 50.8972,
78
+ "eval_rouge2": 30.6779,
79
+ "eval_rougeL": 40.4662,
80
+ "eval_rougeLsum": 45.3636,
81
+ "eval_runtime": 175.4359,
82
+ "eval_samples_per_second": 2.029,
83
+ "eval_steps_per_second": 0.507,
84
+ "step": 2000
85
+ },
86
+ {
87
+ "epoch": 3.0,
88
+ "step": 2403,
89
+ "total_flos": 1.2340414682038272e+16,
90
+ "train_loss": 1.0117533442084115,
91
+ "train_runtime": 1656.9583,
92
+ "train_samples_per_second": 5.801,
93
+ "train_steps_per_second": 1.45
94
+ }
95
+ ],
96
+ "max_steps": 2403,
97
+ "num_train_epochs": 3,
98
+ "total_flos": 1.2340414682038272e+16,
99
+ "trial_name": null,
100
+ "trial_params": null
101
+ }
distilbart_summarizer/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6be8f3d3cae48a85c52142f03b0b5e203369cdeaa56919841e1932bbec733a8c
3
+ size 3771
distilbart_summarizer/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
glove.6B.100d.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95dde4dfd627ab26608d33e76d1195ec059734bd29089ea52cadb08d07c64544
3
+ size 347116733
sentence_bert_disco/.DS_Store ADDED
Binary file (6.15 kB). View file
 
sentence_bert_disco/1_Pooling/config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false
7
+ }
sentence_bert_disco/README.md ADDED
@@ -0,0 +1,125 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: sentence-similarity
3
+ tags:
4
+ - sentence-transformers
5
+ - feature-extraction
6
+ - sentence-similarity
7
+ - transformers
8
+ ---
9
+
10
+ # {MODEL_NAME}
11
+
12
+ This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
13
+
14
+ <!--- Describe your model here -->
15
+
16
+ ## Usage (Sentence-Transformers)
17
+
18
+ Using this model becomes easy when you have [sentence-transformers](https://www.SBERT.net) installed:
19
+
20
+ ```
21
+ pip install -U sentence-transformers
22
+ ```
23
+
24
+ Then you can use the model like this:
25
+
26
+ ```python
27
+ from sentence_transformers import SentenceTransformer
28
+ sentences = ["This is an example sentence", "Each sentence is converted"]
29
+
30
+ model = SentenceTransformer('{MODEL_NAME}')
31
+ embeddings = model.encode(sentences)
32
+ print(embeddings)
33
+ ```
34
+
35
+
36
+
37
+ ## Usage (HuggingFace Transformers)
38
+ Without [sentence-transformers](https://www.SBERT.net), you can use the model like this: First, you pass your input through the transformer model, then you have to apply the right pooling-operation on-top of the contextualized word embeddings.
39
+
40
+ ```python
41
+ from transformers import AutoTokenizer, AutoModel
42
+ import torch
43
+
44
+
45
+ #Mean Pooling - Take attention mask into account for correct averaging
46
+ def mean_pooling(model_output, attention_mask):
47
+ token_embeddings = model_output[0] #First element of model_output contains all token embeddings
48
+ input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
49
+ return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
50
+
51
+
52
+ # Sentences we want sentence embeddings for
53
+ sentences = ['This is an example sentence', 'Each sentence is converted']
54
+
55
+ # Load model from HuggingFace Hub
56
+ tokenizer = AutoTokenizer.from_pretrained('{MODEL_NAME}')
57
+ model = AutoModel.from_pretrained('{MODEL_NAME}')
58
+
59
+ # Tokenize sentences
60
+ encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
61
+
62
+ # Compute token embeddings
63
+ with torch.no_grad():
64
+ model_output = model(**encoded_input)
65
+
66
+ # Perform pooling. In this case, mean pooling.
67
+ sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
68
+
69
+ print("Sentence embeddings:")
70
+ print(sentence_embeddings)
71
+ ```
72
+
73
+
74
+
75
+ ## Evaluation Results
76
+
77
+ <!--- Describe how your model was evaluated -->
78
+
79
+ For an automated evaluation of this model, see the *Sentence Embeddings Benchmark*: [https://seb.sbert.net](https://seb.sbert.net?model_name={MODEL_NAME})
80
+
81
+
82
+ ## Training
83
+ The model was trained with the parameters:
84
+
85
+ **DataLoader**:
86
+
87
+ `torch.utils.data.dataloader.DataLoader` of length 2807 with parameters:
88
+ ```
89
+ {'batch_size': 8, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
90
+ ```
91
+
92
+ **Loss**:
93
+
94
+ `sentence_transformers.losses.SoftmaxLoss.SoftmaxLoss`
95
+
96
+ Parameters of the fit()-Method:
97
+ ```
98
+ {
99
+ "epochs": 5,
100
+ "evaluation_steps": 50000,
101
+ "evaluator": "NoneType",
102
+ "max_grad_norm": 1,
103
+ "optimizer_class": "<class 'transformers.optimization.AdamW'>",
104
+ "optimizer_params": {
105
+ "lr": 3e-05
106
+ },
107
+ "scheduler": "WarmupLinear",
108
+ "steps_per_epoch": null,
109
+ "warmup_steps": 1404,
110
+ "weight_decay": 0.01
111
+ }
112
+ ```
113
+
114
+
115
+ ## Full Model Architecture
116
+ ```
117
+ SentenceTransformer(
118
+ (0): Transformer({'max_seq_length': 170, 'do_lower_case': False}) with Transformer model: BertModel
119
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
120
+ )
121
+ ```
122
+
123
+ ## Citing & Authors
124
+
125
+ <!--- Describe where people can find more information -->
sentence_bert_disco/config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-cased",
3
+ "architectures": [
4
+ "BertModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-12,
15
+ "max_position_embeddings": 512,
16
+ "model_type": "bert",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 12,
19
+ "pad_token_id": 0,
20
+ "position_embedding_type": "absolute",
21
+ "torch_dtype": "float32",
22
+ "transformers_version": "4.18.0",
23
+ "type_vocab_size": 2,
24
+ "use_cache": true,
25
+ "vocab_size": 28996
26
+ }