From 0e9825e2527dfbf1d51520a3ec88416327bb750e Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sun, 14 Jul 2019 23:43:28 +0200 Subject: [PATCH] small fix to run_glue --- examples/run_glue.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/run_glue.py b/examples/run_glue.py index ea5cc9f42d..979c644471 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -166,6 +166,9 @@ def train(args, train_dataset, model, tokenizer): train_iterator.close() break + if args.local_rank in [-1, 0]: + tb_writer.close() + 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])) writer.write("%s = %s\n" % (key, str(result[key]))) - if args.local_rank in [-1, 0]: - tb_writer.close() - return results