Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix issue that pytorch 1.1 does not flush tensorboard

Browse Source
This commit is contained in:
Yin Li 2019-12-12 15:25:45 -05:00
parent bd3798222a
commit 341bdbff84
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/train.py
View File

@ -134,7 +134,7 @@ def gpu_worker(local_rank, args):
if args.rank == 0: if args.rank == 0:
print(end='', flush=True) print(end='', flush=True)
args.logger.flush() args.logger.close()
state = { state = {
'epoch': epoch + 1, 'epoch': epoch + 1,