"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "cd65c41a83da9121f8260490584d9a21b47cc4d8"
Commit c8cba677 authored by thomwolf's avatar thomwolf
Browse files

clean up readme and examples

parent 757750d6
...@@ -362,11 +362,8 @@ python run_classifier.py \ ...@@ -362,11 +362,8 @@ python run_classifier.py \
--task_name MRPC \ --task_name MRPC \
--do_train \ --do_train \
--do_eval \ --do_eval \
--do_lower_case \
--data_dir $GLUE_DIR/MRPC/ \ --data_dir $GLUE_DIR/MRPC/ \
--vocab_file $BERT_BASE_DIR/vocab.txt \ --bert_model bert-base-uncased \
--bert_config_file $BERT_BASE_DIR/bert_config.json \
--init_checkpoint $BERT_PYTORCH_DIR/pytorch_model.bin \
--max_seq_length 128 \ --max_seq_length 128 \
--train_batch_size 32 \ --train_batch_size 32 \
--learning_rate 2e-5 \ --learning_rate 2e-5 \
...@@ -388,12 +385,9 @@ The data for SQuAD can be downloaded with the following links and should be save ...@@ -388,12 +385,9 @@ The data for SQuAD can be downloaded with the following links and should be save
export SQUAD_DIR=/path/to/SQUAD export SQUAD_DIR=/path/to/SQUAD
python run_squad.py \ python run_squad.py \
--vocab_file $BERT_BASE_DIR/vocab.txt \ --bert_model bert-base-uncased \
--bert_config_file $BERT_BASE_DIR/bert_config.json \
--init_checkpoint $BERT_PYTORCH_DIR/pytorch_model.bin \
--do_train \ --do_train \
--do_predict \ --do_predict \
--do_lower_case \
--train_file $SQUAD_DIR/train-v1.1.json \ --train_file $SQUAD_DIR/train-v1.1.json \
--predict_file $SQUAD_DIR/dev-v1.1.json \ --predict_file $SQUAD_DIR/dev-v1.1.json \
--train_batch_size 12 \ --train_batch_size 12 \
......
...@@ -351,11 +351,6 @@ def main(): ...@@ -351,11 +351,6 @@ def main():
type=str, type=str,
required=True, required=True,
help="The name of the task to train.") help="The name of the task to train.")
parser.add_argument("--vocab_file",
default=None,
type=str,
required=True,
help="The vocabulary file that the BERT model was trained on.")
parser.add_argument("--output_dir", parser.add_argument("--output_dir",
default=None, default=None,
type=str, type=str,
......
...@@ -2,7 +2,7 @@ from setuptools import find_packages, setup ...@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
setup( setup(
name="pytorch_pretrained_bert", name="pytorch_pretrained_bert",
version="0.1.1", version="0.1.2",
author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors", author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors",
author_email="thomas@huggingface.co", author_email="thomas@huggingface.co",
description="PyTorch version of Google AI BERT model with script to load Google pre-trained models", description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",
......
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