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

small fix to run_glue

Browse Source
This commit is contained in:
thomwolf
2019-07-14 23:43:28 +02:00
parent 2397f958f9
commit 0e9825e252
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/run_glue.py
View File

@@ -166,6 +166,9 @@ def train(args, train_dataset, model, tokenizer):
train_iterator.close() train_iterator.close()
break break
if args.local_rank in [-1, 0]:
tb_writer.close()
return global_step, tr_loss / global_step return global_step, tr_loss / global_step
@@ -230,9 +233,6 @@ def evaluate(args, model, tokenizer, prefix=""):
logger.info(" %s = %s", key, str(result[key])) logger.info(" %s = %s", key, str(result[key]))
writer.write("%s = %s\n" % (key, str(result[key]))) writer.write("%s = %s\n" % (key, str(result[key])))
if args.local_rank in [-1, 0]:
tb_writer.close()
return results return results