浏览代码

[skip ci] adding horovod rank

[skip ci] fix typo
/distributed-training
Anupam Bhatnagar 5 年前
当前提交
76f9b219
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 7
      ml-agents/mlagents/trainers/stats.py

7
ml-agents/mlagents/trainers/stats.py


from mlagents_envs.logging_util import get_logger
from mlagents_envs.timers import set_gauge
from mlagents.tf_utils import tf, generate_session_config
import horovod.tensorflow as hvd
logger = get_logger(__name__)

) -> None:
is_training = "Not Training."
if "Is Training" in values:
stats_summary = stats_summary = values["Is Training"]
stats_summary = values["Is Training"]
rank = hvd.rank()
"{}: Step: {}. "
"Horovod Rank: {}, {}: Step: {}. "
rank,
category,
step,
time.time() - self.training_start_time,

正在加载...
取消
保存