From b6f365a8ed32eca20034084f74450723414b5de6 Mon Sep 17 00:00:00 2001 From: daniel-shan Date: Mon, 8 Jun 2020 02:36:09 -0700 Subject: [PATCH] Updates args in tf squad example. (#4820) Co-authored-by: Daniel Shan --- examples/question-answering/README.md | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/examples/question-answering/README.md b/examples/question-answering/README.md index d524957b1b..42405e8b59 100644 --- a/examples/question-answering/README.md +++ b/examples/question-answering/README.md @@ -165,17 +165,15 @@ Larger batch size may improve the performance while costing more memory. python run_tf_squad.py \ --model_name_or_path bert-base-uncased \ --output_dir model \ - --max-seq-length 384 \ + --max_seq_length 384 \ --num_train_epochs 2 \ --per_gpu_train_batch_size 8 \ --per_gpu_eval_batch_size 16 \ --do_train \ - --logging_dir logs \ - --mode question-answering \ + --logging_dir logs \ --logging_steps 10 \ --learning_rate 3e-5 \ - --doc_stride 128 \ - --optimizer_name adamw + --doc_stride 128 ``` -For the moment the evaluation is not available in the Tensorflow Trainer only the training. \ No newline at end of file +For the moment the evaluation is not available in the Tensorflow Trainer only the training.