Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
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 \
...
@@ -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
\
...
...
examples/run_classifier.py
View file @
c8cba677
...
@@ -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
,
...
...
setup.py
View file @
c8cba677
...
@@ -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"
,
...
...
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