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
Hide 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.
\ No newline at end of file
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