浏览代码

[skip ci] Revert adding horovod rank in log messages

This reverts commit 04feeae774c081b2faf1a50cc2ea75049f21f12c.
/bug-failed-api-check
Anupam Bhatnagar 5 年前
当前提交
e5b26708
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 4
      ml-agents/mlagents/logging_util.py

4
ml-agents/mlagents/logging_util.py


import logging
import horovod.tensorflow as hvd
horovod_worker = "worker-%s " % str(hvd.rank())
log_format = horovod_worker + "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
log_format = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
logging.basicConfig(level=log_level, format=log_format, datefmt=date_format)
logger = logging.getLogger(name=name)
正在加载...
取消
保存