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
7a9aae10
Commit
7a9aae10
authored
Nov 07, 2019
by
Adrian Bauer
Committed by
Lysandre Debut
Nov 08, 2019
Browse files
Fix run_bertology.py
Make imports and args.overwrite_cache match run_glue.py
parent
1c542df7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
2 deletions
+5
-2
examples/run_bertology.py
examples/run_bertology.py
+5
-2
No files found.
examples/run_bertology.py
View file @
7a9aae10
...
...
@@ -39,8 +39,9 @@ from transformers import (WEIGHTS_NAME,
from
run_glue
import
set_seed
,
load_and_cache_examples
,
ALL_MODELS
,
MODEL_CLASSES
from
utils_glue
import
(
compute_metrics
,
convert_examples_to_features
,
output_modes
,
processors
)
from
transformers
import
glue_compute_metrics
as
compute_metrics
from
transformers
import
glue_output_modes
as
output_modes
from
transformers
import
glue_processors
as
processors
logger
=
logging
.
getLogger
(
__name__
)
...
...
@@ -233,6 +234,8 @@ def main():
help
=
"If > 0: limit the data to a subset of data_subset instances."
)
parser
.
add_argument
(
"--overwrite_output_dir"
,
action
=
'store_true'
,
help
=
"Whether to overwrite data in output directory"
)
parser
.
add_argument
(
'--overwrite_cache'
,
action
=
'store_true'
,
help
=
"Overwrite the cached training and evaluation sets"
)
parser
.
add_argument
(
"--dont_normalize_importance_by_layer"
,
action
=
'store_true'
,
help
=
"Don't normalize importance score by layers"
)
...
...
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