riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

updates to run_glue

Browse Source
This commit is contained in:
thomwolf 2019-07-05 17:22:15 +02:00
parent 99b90edab1
commit 3d5f291386
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/run_glue.py
View File

@ -213,11 +213,12 @@ def load_and_cache_examples(args, task, tokenizer, eval=False):
str(task)))
if os.path.exists(cached_features_file):
logger.info("Loading features from cached file %s", cached_features_file)
features = torch.load(cached_features_file)
else:
features = convert_examples_to_features(examples, label_list, args.max_seq_length, tokenizer, output_mode)
if args.local_rank == -1 or torch.distributed.get_rank() == 0:
logger.info(" Saving eval features into cached file %s", cached_features_file)
logger.info("Saving features into cached file %s", cached_features_file)
torch.save(features, cached_features_file)
return features