saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

save implementation for a2c needed before uncommenting save and load test in test_run.py::test_onpolicy

Browse source
This commit is contained in:
Noah Dormann 2019-11-21 14:44:02 +01:00
parent fb5f192fc4
commit 17f84053b3
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_run.py
View file

@ -32,8 +32,8 @@ def test_cemrl():
def test_onpolicy(model_class, env_id):
model = model_class('MlpPolicy', env_id, policy_kwargs=dict(net_arch=[16]), verbose=1, create_eval_env=True)
model.learn(total_timesteps=1000, eval_freq=500)
model.save("test_save")
model.load("test_save")
#model.save("test_save")
#model.load("test_save")
#os.remove("test_save.pth")