Skip to content

GitLab

Commit f620075d authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix flags in the instructions. 

PiperOrigin-RevId: 332918217
parent 6f832c92
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
official/nlp/nhnet/README.md
View file @ f620075d
......@@ -104,14 +104,13 @@ Please first install TensorFlow 2 and Tensorflow Model Garden following the
```shell
$ python3 trainer.py \
--mode=train_and_eval \
--vocab=/path/to/bert_checkpoint/vocab.txt \
--init_checkpoint=/path/to/bert_checkpoint/bert_model.ckpt \
--params_override='init_from_bert2bert=false' \
--train_file_pattern=$DATA_FOLDER/processed/train.tfrecord* \
--model_dir=/path/to/output/model \
--len_title=15 \
--len_passage=200 \
--max_num_articles=5 \
--num_nhnet_articles=5 \
--model_type=nhnet \
--train_batch_size=16 \
--train_steps=10000 \
......@@ -123,14 +122,13 @@ $ python3 trainer.py \
```shell
$ python3 trainer.py \
--mode=train_and_eval \
--vocab=/path/to/bert_checkpoint/vocab.txt \
--init_checkpoint=/path/to/bert_checkpoint/bert_model.ckpt \
--params_override='init_from_bert2bert=false' \
--train_file_pattern=$DATA_FOLDER/processed/train.tfrecord* \
--model_dir=/path/to/output/model \
--len_title=15 \
--len_passage=200 \
--max_num_articles=5 \
--num_nhnet_articles=5 \
--model_type=nhnet \
--train_batch_size=1024 \
--train_steps=10000 \
......
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