config.yaml 517 字节
Newer Older
X
xixiaoyao 已提交
1 2
task_instance: "mrqa, mlm4mrqa, match4mrqa"
target_tag: 1, 0, 0
X
xixiaoyao 已提交
3
mix_ratio: 1.0, 0.5, 0.5
X
xixiaoyao 已提交
4

X
xixiaoyao 已提交
5
save_path: "output_model/secondrun"
X
xixiaoyao 已提交
6

W
wangxiao 已提交
7
backbone: "ernie"
W
wangxiao 已提交
8
backbone_config_path: "../../pretrain/ernie-en-uncased-large/ernie_config.json"
X
xixiaoyao 已提交
9

W
wangxiao 已提交
10
vocab_path: "../../pretrain/ernie-en-uncased-large/vocab.txt"
X
xixiaoyao 已提交
11 12 13
do_lower_case: True
max_seq_len: 512

X
xixiaoyao 已提交
14
batch_size: 4
X
xixiaoyao 已提交
15 16
save_ckpt_every_n_steps: 5
save_infermodel_every_n_steps: 5
X
Xiaoyao Xi 已提交
17
num_epochs: 2
X
xixiaoyao 已提交
18 19 20 21 22
optimizer: "adam"
learning_rate: 3e-5
warmup_proportion: 0.1
weight_decay: 0.1

X
xixiaoyao 已提交
23
print_every_n_steps: 1