diff options
| author |   zhaoyh6 <zhaoyh6@asiainfo.com> | 2021-11-05 10:36:03 +0800 |
|---|---|---|
| committer | zhaoyh6 <zhaoyh6@asiainfo.com> | 2021-11-05 10:36:23 +0800 |
| commit | 6a725fc388cbb4bf48a044fa281a649fc4eaeb16 (patch) | |
| tree | d471a7f961b0b7a6b963729e151beb3fb54db860 | |
| parent | 6b7fa8fcf82374394ed4116f0974e5326b2a5c9c (diff) | |
fix:Modify NLP offline bug1.0.2
Issue-ID: USECASEUI-605
Signed-off-by: zhaoyh6 <zhaoyh6@asiainfo.com>
Change-Id: Ic458905198f782b72ccad8551e123fc9265f9ac4
| -rw-r--r-- | nlp/scripts/api_squad_offline.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nlp/scripts/api_squad_offline.py b/nlp/scripts/api_squad_offline.py index a54ab7f..87b01c4 100644 --- a/nlp/scripts/api_squad_offline.py +++ b/nlp/scripts/api_squad_offline.py @@ -62,7 +62,7 @@ def serving_input_fn(): def main(FLAGS_output_dir, FLAGS_init_checkpoint_squad, FLAGS_export_dir, FLAGS_predict_file=None, FLAGS_train_file=None, FLAGS_do_predict=False, FLAGS_do_train=False, FLAGS_train_batch_size=16, FLAGS_predict_batch_size=8, FLAGS_learning_rate=5e-5, FLAGS_num_train_epochs=3.0, - FLAGS_max_answer_length=100, FLAGS_max_query_length=64, FLAGS_version_2_with_negative=False,questions=[]]): + FLAGS_max_answer_length=100, FLAGS_max_query_length=64, FLAGS_version_2_with_negative=False,questions=[]): tf.logging.set_verbosity(tf.logging.INFO) bert_config = modeling.BertConfig.from_json_file(FLAGS_bert_config_file) |