Merge pull request #20 from Shujian2015/patch-1

Change model_config.json to xlnet_config.json
zihangdai · Jun 22, 2019 · d93b0f3 · d93b0f3
2 parents 23728ae + 1ea45dd
commit d93b0f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -122,7 +122,7 @@ From here on, we assume XLNet-Large and XLNet-base has been downloaded to `$LARG
  --model_dir=exp/sts-b \
  --uncased=False \
  --spiece_model_file=${LARGE_DIR}/spiece.model \
- --model_config_path=${LARGE_DIR}/model_config.json \
+ --model_config_path=${LARGE_DIR}/xlnet_config.json \
  --init_checkpoint=${LARGE_DIR}/xlnet_model.ckpt \
  --max_seq_length=128 \
  --train_batch_size=8 \
@@ -147,7 +147,7 @@ From here on, we assume XLNet-Large and XLNet-base has been downloaded to `$LARG
  --model_dir=exp/sts-b \
  --uncased=False \
  --spiece_model_file=${LARGE_DIR}/spiece.model \
- --model_config_path=${LARGE_DIR}/model_config.json \
+ --model_config_path=${LARGE_DIR}/xlnet_config.json \
  --max_seq_length=128 \
  --eval_batch_size=8 \
  --num_hosts=1 \