浏览代码

reduce recurrent step/increase batch size

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

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


@pytest.mark.parametrize("use_discrete", [True, False])
def test_recurrent_sac(use_discrete):
step_size = 0.2 if use_discrete else 0.8
step_size = 0.2 if use_discrete else 0.5
env = MemoryEnvironment(
[BRAIN_NAME], use_discrete=use_discrete, step_size=step_size
)

)
new_hyperparams = attr.evolve(
SAC_TF_CONFIG.hyperparameters,
batch_size=128,
batch_size=256,
learning_rate=1e-3,
buffer_init_steps=1000,
steps_per_update=2,

hyperparameters=new_hyperparams,
network_settings=new_networksettings,
max_steps=4000,
max_steps=2000,
framework=FrameworkType.TENSORFLOW,
)
_check_environment_trains(env, {BRAIN_NAME: config})

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


)
new_hyperparams = attr.evolve(
SAC_TORCH_CONFIG.hyperparameters,
batch_size=128,
batch_size=256,
learning_rate=1e-3,
buffer_init_steps=1000,
steps_per_update=2,

hyperparameters=new_hyperparams,
network_settings=new_networksettings,
max_steps=4000,
max_steps=2000,
)
check_environment_trains(env, {BRAIN_NAME: config})

正在加载...
取消
保存