提交 1b7b7690 编写于 作者: Y Yibing Liu

tiny fixes

上级 9eb21f38
...@@ -199,7 +199,6 @@ python -u run_squad.py --use_cuda true\ ...@@ -199,7 +199,6 @@ python -u run_squad.py --use_cuda true\
--do_predict true \ --do_predict true \
--save_steps 100 \ --save_steps 100 \
--warmup_proportion 0.1 \ --warmup_proportion 0.1 \
--validation_steps 100 \
--weight_decay 0.01 \ --weight_decay 0.01 \
--epoch 2 \ --epoch 2 \
--max_seq_len 384 \ --max_seq_len 384 \
...@@ -207,6 +206,7 @@ python -u run_squad.py --use_cuda true\ ...@@ -207,6 +206,7 @@ python -u run_squad.py --use_cuda true\
--predict_file ${SQUAD_PATH}/dev-v1.1.json \ --predict_file ${SQUAD_PATH}/dev-v1.1.json \
--do_lower_case true \ --do_lower_case true \
--doc_stride 128 \ --doc_stride 128 \
--n_best_size 20 \
--train_file ${SQUAD_PATH}/train-v1.1.json \ --train_file ${SQUAD_PATH}/train-v1.1.json \
--learning_rate 3e-5 \ --learning_rate 3e-5 \
--lr_scheduler linear_warmup_decay \ --lr_scheduler linear_warmup_decay \
...@@ -243,7 +243,6 @@ python -u run_squad.py --use_cuda true \ ...@@ -243,7 +243,6 @@ python -u run_squad.py --use_cuda true \
--do_predict true \ --do_predict true \
--save_steps 100 \ --save_steps 100 \
--warmup_proportion 0.1 \ --warmup_proportion 0.1 \
--validation_steps 100 \
--weight_decay 0.01 \ --weight_decay 0.01 \
--epoch 2 \ --epoch 2 \
--max_seq_len 384 \ --max_seq_len 384 \
......
...@@ -22,6 +22,7 @@ import time ...@@ -22,6 +22,7 @@ import time
import argparse import argparse
import numpy as np import numpy as np
import paddle.fluid as fluid import paddle.fluid as fluid
import multiprocessing
import reader.cls as reader import reader.cls as reader
from model.bert import BertConfig from model.bert import BertConfig
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册