浏览代码

[skip ci] adding horovod rank in log messages

/bug-failed-api-check
Anupam Bhatnagar 5 年前
当前提交
56ffcedd
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/logging_util.py

3
ml-agents/mlagents/logging_util.py


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