浏览代码

fix tensorflow test simple rl

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

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


@pytest.mark.parametrize("action_sizes", [(0, 2), (2, 0)])
def test_2d_sac(action_sizes):
env = SimpleEnvironment(
[BRAIN_NAME], action_sizes=action_sizes, action_size=2, step_size=0.8
)
env = SimpleEnvironment([BRAIN_NAME], action_sizes=action_sizes, step_size=0.8)
new_hyperparams = attr.evolve(SAC_TF_CONFIG.hyperparameters, buffer_init_steps=2000)
config = attr.evolve(
SAC_TF_CONFIG,

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


def test_visual_advanced_ppo(vis_encode_type, num_visual):
env = SimpleEnvironment(
[BRAIN_NAME],
action_sizes=True,
action_sizes=(0, 1),
num_visual=num_visual,
num_vector=0,
step_size=0.5,

def test_visual_advanced_sac(vis_encode_type, num_visual):
env = SimpleEnvironment(
[BRAIN_NAME],
action_sizes=True,
action_sizes=(0, 1),
num_visual=num_visual,
num_vector=0,
step_size=0.5,

正在加载...
取消
保存