|
|
|
|
|
|
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)]) |
|
|
|