浏览代码

experiment with 1/1 test

/develop/hybrid-actions
Andrew Cohen 4 年前
当前提交
63757004
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 4
      ml-agents/mlagents/trainers/tests/torch/test_simple_rl.py

4
ml-agents/mlagents/trainers/tests/torch/test_simple_rl.py


# _check_environment_trains(env, {BRAIN_NAME: config}, success_threshold=5.0)
def test_2dhybrid_ppo():
env = HybridEnvironment([BRAIN_NAME], continuous_action_size=1, discrete_action_size=2, step_size=0.8)
env = HybridEnvironment([BRAIN_NAME], continuous_action_size=1, discrete_action_size=1, step_size=0.8)
config = attr.evolve(PPO_CONFIG, hyperparameters=new_hyperparams, max_steps=100000)
config = attr.evolve(PPO_CONFIG, hyperparameters=new_hyperparams, max_steps=10000)
_check_environment_trains(env, {BRAIN_NAME: config}, success_threshold=5.0)
#

正在加载...
取消
保存