in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]
def rank0log(logger, *args, **kwargs):
if hvd.rank() == 0:
if logger:
logger.info("".join([str(x) for x in list(args)]))
else:
print(*args, **kwargs)