Bug fix in saving a2c model.

2017-09-12 02:32:31 +08:00
parent 699919f1cf
commit 3d1e171b3a
1 changed files with 1 additions and 1 deletions
--- a/baselines/a2c/a2c.py
+++ b/baselines/a2c/a2c.py
@@ -69,7 +69,7 @@ class Model(object):

        def save(save_path):
            ps = sess.run(params)
-            make_path(save_path)
+            make_path(osp.dirname(save_path))
            joblib.dump(ps, save_path)

        def load(load_path):