Skip to content

GitLab

Commit b429689b authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Documentation minor update. 

PiperOrigin-RevId: 412943633
parent 85d08132
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/docs/train.md
View file @ b429689b
......@@ -113,6 +113,7 @@ python3 train.py \
--experiment=bert/sentence_prediction \
--mode=train_and_eval \
--model_dir=$OUTPUT_DIR \
--config_file=configs/models/bert_en_uncased_base.yaml \
--config_file=configs/experiments/glue_mnli_matched.yaml \
--tfhub_cache_dir=$OUTPUT_DIR/hub_cache \
--tpu=${TPU_NAME} \
......@@ -172,6 +173,7 @@ python3 train.py \
--experiment=bert/squad \
--mode=train_and_eval \
--model_dir=$OUTPUT_DIR \
--config_file=configs/models/bert_en_uncased_base.yaml \
--config_file=configs/experiments/squad_v1.1.yaml \
--tpu=${TPU_NAME} \
--params_override=$PARAMS
......
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