SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix run_bertology.py

Browse Source 
Make imports and args.overwrite_cache match run_glue.py
This commit is contained in:
Adrian Bauer
2019-11-07 17:08:39 -05:00
committed by Lysandre Debut
parent 1c542df7e5
commit 7a9aae1044
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/run_bertology.py
View File

@@ -39,8 +39,9 @@ from transformers import (WEIGHTS_NAME,
from run_glue import set_seed, load_and_cache_examples, ALL_MODELS, MODEL_CLASSES from run_glue import set_seed, load_and_cache_examples, ALL_MODELS, MODEL_CLASSES
from utils_glue import (compute_metrics, convert_examples_to_features, from transformers import glue_compute_metrics as compute_metrics
output_modes, processors) from transformers import glue_output_modes as output_modes
from transformers import glue_processors as processors
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -233,6 +234,8 @@ def main():
help="If > 0: limit the data to a subset of data_subset instances.") help="If > 0: limit the data to a subset of data_subset instances.")
parser.add_argument("--overwrite_output_dir", action='store_true', parser.add_argument("--overwrite_output_dir", action='store_true',
help="Whether to overwrite data in output directory") 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', parser.add_argument("--dont_normalize_importance_by_layer", action='store_true',
help="Don't normalize importance score by layers") help="Don't normalize importance score by layers")