浏览代码

-

/exp-vince
vincentpierre 4 年前
当前提交
d2ad8f12
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/simple_env_manager.py

3
ml-agents/mlagents/trainers/simple_env_manager.py


from mlagents_envs.side_channel.environment_parameters_channel import (
EnvironmentParametersChannel,
)
import numpy as np
class SimpleEnvManager(EnvManager):

self.previous_all_action_info = all_action_info
for brain_name, action_info in all_action_info.items():
self.env.set_actions(brain_name, action_info.action)
self.env.set_actions(brain_name, np.array(action_info.action))
self.env.step()
all_step_result = self._generate_all_results()

正在加载...
取消
保存