upload fp16 models
Browse files- added_tokens.json +1 -0
- config.json +41 -0
- pytorch_model.bin +3 -0
- special_tokens_map.json +1 -0
- spm.model +3 -0
- tokenizer_config.json +1 -0
- training_config.yaml +96 -0
added_tokens.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"[MASK]": 128000}
|
config.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "pretrained-models/deberta-v2-xxlarge",
|
| 3 |
+
"add_enhanced_decoder": true,
|
| 4 |
+
"architectures": [
|
| 5 |
+
"DebertaV2ForMultipleChoicePreTrain"
|
| 6 |
+
],
|
| 7 |
+
"attention_head_size": 64,
|
| 8 |
+
"attention_probs_dropout_prob": 0.1,
|
| 9 |
+
"conv_act": "gelu",
|
| 10 |
+
"conv_kernel_size": 3,
|
| 11 |
+
"hidden_act": "gelu",
|
| 12 |
+
"hidden_dropout_prob": 0.1,
|
| 13 |
+
"hidden_size": 1536,
|
| 14 |
+
"initializer_range": 0.02,
|
| 15 |
+
"intermediate_size": 6144,
|
| 16 |
+
"layer_norm_eps": 1e-07,
|
| 17 |
+
"max_position_embeddings": 512,
|
| 18 |
+
"max_relative_positions": -1,
|
| 19 |
+
"mlp_hidden_size": 3072,
|
| 20 |
+
"model_type": "deberta-v2",
|
| 21 |
+
"norm_rel_ebd": "layer_norm",
|
| 22 |
+
"num_attention_heads": 24,
|
| 23 |
+
"num_hidden_layers": 48,
|
| 24 |
+
"pad_token_id": 0,
|
| 25 |
+
"pooler_dropout": 0,
|
| 26 |
+
"pooler_hidden_act": "gelu",
|
| 27 |
+
"pooler_hidden_size": 1536,
|
| 28 |
+
"pos_att_type": [
|
| 29 |
+
"p2c",
|
| 30 |
+
"c2p"
|
| 31 |
+
],
|
| 32 |
+
"position_biased_input": false,
|
| 33 |
+
"position_buckets": 256,
|
| 34 |
+
"relative_attention": true,
|
| 35 |
+
"share_att_key": true,
|
| 36 |
+
"torch_dtype": "float32",
|
| 37 |
+
"transformers_version": "4.15.0",
|
| 38 |
+
"type_vocab_size": 0,
|
| 39 |
+
"use_stable_embedding": false,
|
| 40 |
+
"vocab_size": 128100
|
| 41 |
+
}
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:74d9f7ed01565f457923e0c26f1a86c21ffd30e7bfa694eded38971658b75639
|
| 3 |
+
size 3539235155
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"bos_token": "[CLS]", "eos_token": "[SEP]", "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
|
spm.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5598d5e96f339a8d980c15f9afd405a2e5e1be7db41de3ed13b0f03fac1e8c17
|
| 3 |
+
size 2447305
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"do_lower_case": false, "bos_token": "[CLS]", "eos_token": "[SEP]", "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "split_by_punct": false, "sp_model_kwargs": {}, "vocab_type": "spm", "special_tokens_map_file": null, "tokenizer_file": null, "name_or_path": "pretrained-models/deberta-v2-xxlarge", "tokenizer_class": "DebertaV2Tokenizer"}
|
training_config.yaml
ADDED
|
@@ -0,0 +1,96 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
train_file: wiki_erica_path/v7/union/train_distant.path_v7.train.0.pkl
|
| 2 |
+
dev_file: wiki_erica_path/v7/union/train_distant.path_v7.dev.pkl
|
| 3 |
+
test_file: null
|
| 4 |
+
model:
|
| 5 |
+
_target_: models.deberta.DebertaV2ForMultipleChoicePreTrain.from_pretrained
|
| 6 |
+
mlp_hidden_size: 3072
|
| 7 |
+
fs_checkpoint: false
|
| 8 |
+
fs_checkpoint_offload_to_cpu: false
|
| 9 |
+
read_tensor:
|
| 10 |
+
_target_: dataset.wiki_entity_path_v8_2.convert_examples_into_features
|
| 11 |
+
max_neg_num: 3
|
| 12 |
+
aug_num: 1
|
| 13 |
+
max_seq_length: 256
|
| 14 |
+
shuffle_context: true
|
| 15 |
+
min_rep_num: 5
|
| 16 |
+
geo_p: 0.4
|
| 17 |
+
deduct_ratio: 1.0
|
| 18 |
+
context_ratio: 1.0
|
| 19 |
+
num_workers: 32
|
| 20 |
+
extended_vocab: null
|
| 21 |
+
collator:
|
| 22 |
+
_target_: dataset.wiki_entity_path_v8.WikiPathDatasetCollatorWithContext
|
| 23 |
+
max_seq_length: 256
|
| 24 |
+
tokenizer: pretrained-models/deberta-v2-xxlarge
|
| 25 |
+
mlm_probability: 0.15
|
| 26 |
+
max_option_num: 4
|
| 27 |
+
swap: true
|
| 28 |
+
num_workers: 4
|
| 29 |
+
prefetch_factor: 4
|
| 30 |
+
model_name_or_path: pretrained-models/deberta-v2-xxlarge
|
| 31 |
+
pretrain: null
|
| 32 |
+
output_dir: experiments/deberta.v2.xxlarge.path.v7_v8.2.2.1aug.ctx.A100.v1.3.w4.s${seed}.fsdp.adamw
|
| 33 |
+
do_train: Train
|
| 34 |
+
evaluate_during_training: true
|
| 35 |
+
do_eval: false
|
| 36 |
+
eval_sub_path: null
|
| 37 |
+
do_preprocess: false
|
| 38 |
+
per_gpu_train_batch_size: 2
|
| 39 |
+
per_gpu_eval_batch_size: 2
|
| 40 |
+
learning_rate: 1.0e-05
|
| 41 |
+
gradient_accumulation_steps: 512
|
| 42 |
+
weight_decay: 0.01
|
| 43 |
+
adam_epsilon: 1.0e-06
|
| 44 |
+
adam_betas: (0.9, 0.999)
|
| 45 |
+
max_grad_norm: 1.0
|
| 46 |
+
num_train_epochs: 1
|
| 47 |
+
max_steps: 200
|
| 48 |
+
warmup_proportion: 0.2
|
| 49 |
+
warmup_steps: 0
|
| 50 |
+
optimizer: null
|
| 51 |
+
use_nvlamb: null
|
| 52 |
+
bit_training: null
|
| 53 |
+
multi_tensor: null
|
| 54 |
+
logging_steps: 1
|
| 55 |
+
save_steps: 50
|
| 56 |
+
eval_steps: 50
|
| 57 |
+
no_cuda: false
|
| 58 |
+
seed: 42
|
| 59 |
+
local_rank: 0
|
| 60 |
+
fp16: true
|
| 61 |
+
fp16_opt_level: O2
|
| 62 |
+
ds_cfg:
|
| 63 |
+
train_micro_batch_size_per_gpu: ${per_gpu_train_batch_size}
|
| 64 |
+
gradient_accumulation_steps: ${gradient_accumulation_steps}
|
| 65 |
+
optimizer:
|
| 66 |
+
type: AdamW
|
| 67 |
+
params:
|
| 68 |
+
lr: ${learning_rate}
|
| 69 |
+
betas:
|
| 70 |
+
- 0.9
|
| 71 |
+
- 0.999
|
| 72 |
+
eps: ${adam_epsilon}
|
| 73 |
+
weight_decay: ${weight_decay}
|
| 74 |
+
scheduler:
|
| 75 |
+
type: WarmupDecayLR
|
| 76 |
+
params:
|
| 77 |
+
total_num_steps: null
|
| 78 |
+
warmup_max_lr: ${learning_rate}
|
| 79 |
+
warmup_num_steps: null
|
| 80 |
+
warmup_type: linear
|
| 81 |
+
gradient_clipping: ${max_grad_norm}
|
| 82 |
+
fp16:
|
| 83 |
+
enabled: ${fp16}
|
| 84 |
+
initial_scale_power: 12
|
| 85 |
+
zero_optimization:
|
| 86 |
+
stage: 3
|
| 87 |
+
steps_per_print: 1024
|
| 88 |
+
reshard_after_forward: false
|
| 89 |
+
flatten_parameters: true
|
| 90 |
+
move_grads_to_cpu: false
|
| 91 |
+
move_params_to_cpu: false
|
| 92 |
+
n_gpu: 1
|
| 93 |
+
device: cuda:0
|
| 94 |
+
train_batch_size: 2
|
| 95 |
+
eval_batch_size: 2
|
| 96 |
+
note: null
|