diff --git a/BERT/README.md b/BERT/README.md index 7feab019fbd369813d706f41abfb6eb1a88edcfc..5de9e769f5e58683452ff6186c3274eedb75c86b 100644 --- a/BERT/README.md +++ b/BERT/README.md @@ -199,7 +199,6 @@ python -u run_squad.py --use_cuda true\ --do_predict true \ --save_steps 100 \ --warmup_proportion 0.1 \ - --validation_steps 100 \ --weight_decay 0.01 \ --epoch 2 \ --max_seq_len 384 \ @@ -207,6 +206,7 @@ python -u run_squad.py --use_cuda true\ --predict_file ${SQUAD_PATH}/dev-v1.1.json \ --do_lower_case true \ --doc_stride 128 \ + --n_best_size 20 \ --train_file ${SQUAD_PATH}/train-v1.1.json \ --learning_rate 3e-5 \ --lr_scheduler linear_warmup_decay \ @@ -243,7 +243,6 @@ python -u run_squad.py --use_cuda true \ --do_predict true \ --save_steps 100 \ --warmup_proportion 0.1 \ - --validation_steps 100 \ --weight_decay 0.01 \ --epoch 2 \ --max_seq_len 384 \ diff --git a/BERT/predict_classifier.py b/BERT/predict_classifier.py index 51e840db1113ec7c4c53c94a1f76000b87756460..2edfb6510be60492ab6d55da8ce229d83c6e978c 100644 --- a/BERT/predict_classifier.py +++ b/BERT/predict_classifier.py @@ -22,6 +22,7 @@ import time import argparse import numpy as np import paddle.fluid as fluid +import multiprocessing import reader.cls as reader from model.bert import BertConfig