Skip to content

GitLab

Unverified Commit b6f365a8 authored by daniel-shan's avatar daniel-shan Committed by GitHub
Browse files

Updates args in tf squad example. (#4820) 


Co-authored-by: default avatarDaniel Shan <daniel.shan@workday.com>
parent e33fdc93
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 6 deletions
examples/question-answering/README.md
View file @ b6f365a8
...@@ -165,17 +165,15 @@ Larger batch size may improve the performance while costing more memory. ...@@ -165,17 +165,15 @@ Larger batch size may improve the performance while costing more memory.
python run_tf_squad.py \ python run_tf_squad.py \
--model_name_or_path bert-base-uncased \ --model_name_or_path bert-base-uncased \
--output_dir model \ --output_dir model \
--max-seq-length 384 \ --max_seq_length 384 \
--num_train_epochs 2 \ --num_train_epochs 2 \
--per_gpu_train_batch_size 8 \ --per_gpu_train_batch_size 8 \
--per_gpu_eval_batch_size 16 \ --per_gpu_eval_batch_size 16 \
--do_train \ --do_train \
--logging_dir logs \ --logging_dir logs \
--mode question-answering \
--logging_steps 10 \ --logging_steps 10 \
--learning_rate 3e-5 \ --learning_rate 3e-5 \
--doc_stride 128 \ --doc_stride 128
--optimizer_name adamw
``` ```
For the moment the evaluation is not available in the Tensorflow Trainer only the training. For the moment the evaluation is not available in the Tensorflow Trainer only the training.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment