config_demo2.yaml 423 字节
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 7

backbone: "ernie"
X
xixiaoyao 已提交
8
backbone_config_path: "pretrain_model/ernie/ernie_config.json"
X
xixiaoyao 已提交
9

X
xixiaoyao 已提交
10
vocab_path: "pretrain_model/ernie/vocab.txt"
X
xixiaoyao 已提交
11 12 13
do_lower_case: True
max_seq_len: 512

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

X
xixiaoyao 已提交
21
print_every_n_steps: 1