浏览代码

fixing ci ppo_policy

/develop/magic-string
Andrew Cohen 5 年前
当前提交
c3a92afa
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/ppo/trainer.py

2
ml-agents/mlagents/trainers/ppo/trainer.py


def create_policy(self, brain_parameters: BrainParameters) -> TFPolicy:
if self.multi_gpu and len(get_devices()) > 1:
self.ppo_policy = MultiGpuPPOPolicy(
self.ppo_policy: PPOPolicy = MultiGpuPPOPolicy(
self.seed,
brain_parameters,
self.trainer_parameters,

正在加载...
取消
保存