|
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize("action_sizes", [(0, 2), (2, 0)]) |
|
|
|
def test_2d_sac(action_sizes): |
|
|
|
env = SimpleEnvironment( |
|
|
|
[BRAIN_NAME], action_sizes=action_sizes, action_size=2, step_size=0.8 |
|
|
|
) |
|
|
|
env = SimpleEnvironment([BRAIN_NAME], action_sizes=action_sizes, step_size=0.8) |
|
|
|
new_hyperparams = attr.evolve(SAC_TF_CONFIG.hyperparameters, buffer_init_steps=2000) |
|
|
|
config = attr.evolve( |
|
|
|
SAC_TF_CONFIG, |
|
|
|