浏览代码

Fixing the tests

/layernorm
vincentpierre 4 年前
当前提交
8dee7970
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 4
      ml-agents/mlagents/trainers/tests/torch/test_hybrid.py

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


config = attr.evolve(
SAC_TORCH_CONFIG, hyperparameters=new_hyperparams, max_steps=2000
)
check_environment_trains(env, {BRAIN_NAME: config}, success_threshold=0.9)
check_environment_trains(
env, {BRAIN_NAME: config}, success_threshold=0.9, training_seed=1336
)
@pytest.mark.parametrize("num_visual", [1, 2])

正在加载...
取消
保存