forked from mrq/DL-Art-School
Fix illegal tb_logger use in distributed training
This commit is contained in:
parent
bba283776c
commit
a7541b6d8d
|
@ -203,6 +203,7 @@ def main():
|
|||
message += ')] '
|
||||
for k, v in logs.items():
|
||||
if 'histogram' in k:
|
||||
if rank <= 0:
|
||||
tb_logger.add_histogram(k, v, current_step)
|
||||
else:
|
||||
message += '{:s}: {:.4e} '.format(k, v)
|
||||
|
|
Loading…
Reference in New Issue
Block a user