浏览代码

adding globals

/MLA-1734-demo-provider
Anupam Bhatnagar 4 年前
当前提交
7a25b882
共有 1 个文件被更改,包括 13 次插入0 次删除
  1. 13
      ml-agents/mlagents/tf_utils/globals.py

13
ml-agents/mlagents/tf_utils/globals.py


try:
import horovod.tensorflow as hvd
except ImportError:
hvd = None
def get_hvd():
return hvd
def get_rank():
rank = hvd.rank() if hvd else None
return rank
正在加载...
取消
保存