提交 74bc3ca6 编写于 作者: H Hongyu Li 提交者: Yibing Liu

Update ERNIE model (#2646)

上级 a6cbcd73
......@@ -15,7 +15,8 @@
# limitations under the License.
# ==============================================================================
# download pre_train model
wget ftp://cp01-rdqa04-dev119-zhangxiyuan01.epc.baidu.com/home/users/zhangxiyuan01/models/ernie_model.tar.gz
tar -xvf ernie_model.tar.gz
rm ernie_model.tar.gz
# download pretrained ERNIE model
wget --no-check-certificate https://ernie.bj.bcebos.com/ERNIE_en_1.0.tgz
tar -xvf ERNIE_en_1.0.tgz
rm ERNIE_en_1.0.tgz
ln -s ERNIE_en_1.0 ernie_model
......@@ -46,7 +46,7 @@ python -u src/run_mrqa.py --use_cuda true\
--weight_decay 0.01 \
--epoch 2 \
--max_seq_len 512 \
--bert_config_path ${ERNIE_BASE_PATH}/bert_config.json \
--bert_config_path ${ERNIE_BASE_PATH}/ernie_config.json \
--predict_file ${DATA_PATH_dev}/mrqa-combined.raw.json \
--do_lower_case true \
--doc_stride 128 \
......
......@@ -48,7 +48,7 @@ python -u src/run_mrqa.py --use_cuda true\
--weight_decay 0.01 \
--epoch 2 \
--max_seq_len 512 \
--bert_config_path ${ERNIE_BASE_PATH}/bert_config.json \
--bert_config_path ${ERNIE_BASE_PATH}/ernie_config.json \
--predict_file ${DATA_PATH_dev}/mrqa-combined.raw.json \
--do_lower_case true \
--doc_stride 128 \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册