diff --git a/demo/demo1/config.yaml b/demo/demo1/config.yaml index 0cca90ff519fc9606214dd3aad4c5763092e9bcc..75b685fc933939f1787c9171ceda3f2ca3cec37b 100644 --- a/demo/demo1/config.yaml +++ b/demo/demo1/config.yaml @@ -2,7 +2,7 @@ task_instance: "mrqa" save_path: "output_model/firstrun" -backbone: "bert-en-uncased-large" +backbone: "bert" backbone_config_path: "../../pretrain/bert-en-uncased-large/bert_config.json" batch_size: 4 diff --git a/demo/demo2/config.yaml b/demo/demo2/config.yaml index 2bd539051b4dbcc7b65763599f69bf81402f6047..95f8dfb91bea9fa9e07a70dfe069a9280f26e26a 100644 --- a/demo/demo2/config.yaml +++ b/demo/demo2/config.yaml @@ -4,7 +4,7 @@ mix_ratio: 1.0, 0.5, 0.5 save_path: "output_model/secondrun" -backbone: "ernie-en-uncased-large" +backbone: "ernie" backbone_config_path: "../../pretrain/ernie-en-uncased-large/ernie_config.json" vocab_path: "../../pretrain/ernie-en-uncased-large/vocab.txt" diff --git a/demo/demo3/config.yaml b/demo/demo3/config.yaml index 0595e58eb2c099807f1a9f790428406f134efe89..23beac1f80203a0a16edf0106ca26b66562f8ef1 100644 --- a/demo/demo3/config.yaml +++ b/demo/demo3/config.yaml @@ -4,7 +4,7 @@ task_reuse_tag: 0,0,1,1,0,2 save_path: "output_model/thirdrun" -backbone: "ernie-en-uncased-large" +backbone: "ernie" backbone_config_path: "../../pretrain/ernie-en-uncased-large/ernie_config.json" vocab_path: "../../pretrain/ernie-en-uncased-large/vocab.txt"