浏览代码

adjust step size

/develop/gail-srl-hack
Andrew Cohen 4 年前
当前提交
e547f26c
共有 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


def test_gail(simple_record, action_sizes, trainer_config):
demo_path = simple_record(action_sizes)
env = SimpleEnvironment(
[BRAIN_NAME], action_sizes=action_sizes, step_size=0.2, gail=True
[BRAIN_NAME], action_sizes=action_sizes, step_size=0.3, gail=True
)
bc_settings = BehavioralCloningSettings(demo_path=demo_path, steps=1000)
reward_signals = {

num_visual=1,
num_vector=0,
action_sizes=action_sizes,
step_size=0.2,
step_size=0.3,
gail=True,
)
bc_settings = BehavioralCloningSettings(demo_path=demo_path, steps=1000)

正在加载...
取消
保存