mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix illegal tb_logger use in distributed training

Browse Source
This commit is contained in:
James Betker 2020-07-23 09:14:01 -06:00
parent bba283776c
commit a7541b6d8d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/train.py
View File

@ -203,7 +203,8 @@ def main():
message += ')] ' message += ')] '
for k, v in logs.items(): for k, v in logs.items():
if 'histogram' in k: if 'histogram' in k:
tb_logger.add_histogram(k, v, current_step) if rank <= 0:
tb_logger.add_histogram(k, v, current_step)
else: else:
message += '{:s}: {:.4e} '.format(k, v) message += '{:s}: {:.4e} '.format(k, v)
# tensorboard logger # tensorboard logger