浏览代码

Change learning rate

/MLA-1734-demo-provider
Arthur Juliani 4 年前
当前提交
b4d8cf54
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 6
      ml-agents/mlagents/trainers/tests/torch/test_simple_rl.py

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


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

network_settings=new_networksettings,
max_steps=3000,
max_steps=4000,
check_environment_trains(env, {BRAIN_NAME: config}, training_seed=1212)
check_environment_trains(env, {BRAIN_NAME: config}, training_seed=1213)
@pytest.mark.parametrize("action_sizes", [(0, 1), (1, 0)])

正在加载...
取消
保存