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

X
xixiaoyao 已提交
5
save_path: "output_model/secondrun"
X
xixiaoyao 已提交
6 7 8 9 10 11 12 13

backbone: "ernie"
backbone_config_path: "pretrain_model/ernie/ernie_config.json"

vocab_path: "pretrain_model/ernie/vocab.txt"
do_lower_case: True
max_seq_len: 512

X
xixiaoyao 已提交
14
batch_size: 4
X
xixiaoyao 已提交
15
num_epochs: 0.5
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