saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-03 03:59:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for SAC with different entropy temperature

Browse source
This commit is contained in:
Antonin Raffin 2019-12-02 11:47:52 +01:00
parent 03a84f97ea
commit 21e655ecbf
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_run.py
View file

@ -37,8 +37,9 @@ def test_onpolicy(model_class, env_id):
# model.load("test_save")
# os.remove("test_save.pth")
def test_sac():
@pytest.mark.parametrize("ent_coef", ['auto', 0.01])
def test_sac(ent_coef):
model = SAC('MlpPolicy', 'Pendulum-v0', policy_kwargs=dict(net_arch=[64, 64]),
learning_starts=100, verbose=1, create_eval_env=True, ent_coef='auto',
learning_starts=100, verbose=1, create_eval_env=True, ent_coef=ent_coef,
action_noise=NormalActionNoise(np.zeros(1), np.zeros(1)))
model.learn(total_timesteps=1000, eval_freq=500)