浏览代码

-

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

10
ml-agents/mlagents/trainers/learn.py


from mlagents.trainers.training_status import GlobalTrainingStatus
from mlagents_envs.base_env import BaseEnv
from mlagents.trainers.subprocess_env_manager import SubprocessEnvManager
from mlagents.trainers.simple_env_manager import SimpleEnvManager
from mlagents_envs.side_channel.side_channel import SideChannel
from mlagents_envs.side_channel.engine_configuration_channel import EngineConfig
from mlagents_envs.timers import (

target_frame_rate=engine_settings.target_frame_rate,
capture_frame_rate=engine_settings.capture_frame_rate,
)
env_manager = SubprocessEnvManager(
env_factory, engine_config, env_settings.num_envs
)
# env_manager = SubprocessEnvManager(
# env_factory, engine_config, env_settings.num_envs
# )
env_manager = SimpleEnvManager(env_factory(0, []), env_parameter_manager)
trainer_factory = TrainerFactory(
trainer_config=options.behaviors,

正在加载...
取消
保存