Bug fix in saving ACER model

2018-03-01 10:24:14 +01:00
parent 6bdf2f55a2
commit 14f2f9328c
1 changed files with 1 additions and 1 deletions
--- a/baselines/acer/acer_simple.py
+++ b/baselines/acer/acer_simple.py
@@ -198,7 +198,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)

        self.train = train