浏览代码

removing horovod import

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

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


try:
import horovod.tensorflow as hvd
except ImportError:
hvd = None
from typing import Optional
def get_hvd():
return hvd
_rank: Optional[int] = None
_local_rank: Optional[int] = None
def get_rank():
rank = hvd.rank() if hvd else None
return rank
def get_rank() -> Optional[int]:
return _rank
正在加载...
取消
保存