浏览代码

branch rebased

/distributed-training
Anupam Bhatnagar 5 年前
当前提交
c76c2b0d
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/trainer/trainer.py

1
ml-agents/mlagents/trainers/trainer/trainer.py


from mlagents.trainers.brain import BrainParameters
from mlagents.trainers.policy import Policy
from mlagents.trainers.exception import UnityTrainerException
from mlagents.trainers.behavior_id_utils import BehaviorIdentifiers
import horovod.tensorflow as hvd

正在加载...
取消
保存