Skip to content

GitLab

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

Fix flags in the instructions. 

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