Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
c8cba677
Commit
c8cba677
authored
Nov 17, 2018
by
thomwolf
Browse files
clean up readme and examples
parent
757750d6
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
14 deletions
+3
-14
README.md
README.md
+2
-8
examples/run_classifier.py
examples/run_classifier.py
+0
-5
setup.py
setup.py
+1
-1
No files found.
README.md
View file @
c8cba677
...
...
@@ -362,11 +362,8 @@ python run_classifier.py \
--task_name
MRPC
\
--do_train
\
--do_eval
\
--do_lower_case
\
--data_dir
$GLUE_DIR
/MRPC/
\
--vocab_file
$BERT_BASE_DIR
/vocab.txt
\
--bert_config_file
$BERT_BASE_DIR
/bert_config.json
\
--init_checkpoint
$BERT_PYTORCH_DIR
/pytorch_model.bin
\
--bert_model
bert-base-uncased
\
--max_seq_length
128
\
--train_batch_size
32
\
--learning_rate
2e-5
\
...
...
@@ -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
python run_squad.py
\
--vocab_file
$BERT_BASE_DIR
/vocab.txt
\
--bert_config_file
$BERT_BASE_DIR
/bert_config.json
\
--init_checkpoint
$BERT_PYTORCH_DIR
/pytorch_model.bin
\
--bert_model
bert-base-uncased
\
--do_train
\
--do_predict
\
--do_lower_case
\
--train_file
$SQUAD_DIR
/train-v1.1.json
\
--predict_file
$SQUAD_DIR
/dev-v1.1.json
\
--train_batch_size
12
\
...
...
examples/run_classifier.py
View file @
c8cba677
...
...
@@ -351,11 +351,6 @@ def main():
type
=
str
,
required
=
True
,
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"
,
default
=
None
,
type
=
str
,
...
...
setup.py
View file @
c8cba677
...
...
@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
setup
(
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_email
=
"thomas@huggingface.co"
,
description
=
"PyTorch version of Google AI BERT model with script to load Google pre-trained models"
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment