Fix issue that pytorch 1.1 does not flush tensorboard

2019-12-12 15:25:45 -05:00 · 2019-12-12 15:25:45 -05:00 · 341bdbff84
commit 341bdbff84
parent bd3798222a
1 changed files with 1 additions and 1 deletions
--- 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,