From 341bdbff848f6518674d9909340bf67f716405fa Mon Sep 17 00:00:00 2001
From: Yin Li <eelregit@gmail.com>
Date: Thu, 12 Dec 2019 15:25:45 -0500
Subject: [PATCH] Fix issue that pytorch 1.1 does not flush tensorboard

---
 map2map/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/map2map/train.py b/map2map/train.py
index d691017..c77fab8 100644
--- a/map2map/train.py
+++ b/map2map/train.py
@@ -134,7 +134,7 @@ def gpu_worker(local_rank, args):
 
         if args.rank == 0:
             print(end='', flush=True)
-            args.logger.flush()
+            args.logger.close()
 
             state = {
                 'epoch': epoch + 1,