Skip to content

GitLab

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

Fix bert readme. Always use keras_bert path. 

PiperOrigin-RevId: 286327318
parent 1722b691
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
official/nlp/bert/README.md
View file @ 83076c16
...@@ -141,7 +141,7 @@ and unpack it to some directory `$GLUE_DIR`. ...@@ -141,7 +141,7 @@ and unpack it to some directory `$GLUE_DIR`.
```shell ```shell
export GLUE_DIR=~/glue export GLUE_DIR=~/glue
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export TASK_NAME=MNLI export TASK_NAME=MNLI
export OUTPUT_DIR=gs://some_bucket/datasets export OUTPUT_DIR=gs://some_bucket/datasets
...@@ -172,7 +172,7 @@ The necessary files can be found here: ...@@ -172,7 +172,7 @@ The necessary files can be found here:
```shell ```shell
export SQUAD_DIR=~/squad export SQUAD_DIR=~/squad
export SQUAD_VERSION=v1.1 export SQUAD_VERSION=v1.1
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export OUTPUT_DIR=gs://some_bucket/datasets export OUTPUT_DIR=gs://some_bucket/datasets
python create_finetuning_data.py \ python create_finetuning_data.py \
...@@ -190,7 +190,7 @@ python create_finetuning_data.py \ ...@@ -190,7 +190,7 @@ python create_finetuning_data.py \
* Cloud Storage * Cloud Storage
The unzipped pre-trained model files can also be found in the Google Cloud The unzipped pre-trained model files can also be found in the Google Cloud
Storage folder `gs://cloud-tpu-checkpoints/bert/tf_20`. For example: Storage folder `gs://cloud-tpu-checkpoints/bert/keras_bert`. For example:
```shell ```shell
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
...@@ -221,7 +221,7 @@ For GPU memory of 16GB or smaller, you may try to use `BERT-Base` ...@@ -221,7 +221,7 @@ For GPU memory of 16GB or smaller, you may try to use `BERT-Base`
(uncased_L-12_H-768_A-12). (uncased_L-12_H-768_A-12).
```shell ```shell
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export MODEL_DIR=gs://some_bucket/my_output_dir export MODEL_DIR=gs://some_bucket/my_output_dir
export GLUE_DIR=gs://some_bucket/datasets export GLUE_DIR=gs://some_bucket/datasets
export TASK=MRPC export TASK=MRPC
...@@ -246,7 +246,7 @@ To use TPU, you only need to switch distribution strategy type to `tpu` with TPU ...@@ -246,7 +246,7 @@ To use TPU, you only need to switch distribution strategy type to `tpu` with TPU
information and use remote storage for model checkpoints. information and use remote storage for model checkpoints.
```shell ```shell
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export TPU_IP_ADDRESS='???' export TPU_IP_ADDRESS='???'
export MODEL_DIR=gs://some_bucket/my_output_dir export MODEL_DIR=gs://some_bucket/my_output_dir
export GLUE_DIR=gs://some_bucket/datasets export GLUE_DIR=gs://some_bucket/datasets
...@@ -278,7 +278,7 @@ For GPU memory of 16GB or smaller, you may try to use `BERT-Base` ...@@ -278,7 +278,7 @@ For GPU memory of 16GB or smaller, you may try to use `BERT-Base`
(uncased_L-12_H-768_A-12). (uncased_L-12_H-768_A-12).
```shell ```shell
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export SQUAD_DIR=gs://some_bucket/datasets export SQUAD_DIR=gs://some_bucket/datasets
export MODEL_DIR=gs://some_bucket/my_output_dir export MODEL_DIR=gs://some_bucket/my_output_dir
export SQUAD_VERSION=v1.1 export SQUAD_VERSION=v1.1
...@@ -302,7 +302,7 @@ To use TPU, you need switch distribution strategy type to `tpu` with TPU ...@@ -302,7 +302,7 @@ To use TPU, you need switch distribution strategy type to `tpu` with TPU
information. information.
```shell ```shell
export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16 export BERT_BASE_DIR=gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-24_H-1024_A-16
export TPU_IP_ADDRESS='???' export TPU_IP_ADDRESS='???'
export MODEL_DIR=gs://some_bucket/my_output_dir export MODEL_DIR=gs://some_bucket/my_output_dir
export SQUAD_DIR=gs://some_bucket/datasets export SQUAD_DIR=gs://some_bucket/datasets
......
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