gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bug fix in saving ACER model

Browse Source
This commit is contained in:
Antonin RAFFIN
2018-03-01 10:24:14 +01:00
committed by GitHub
parent 6bdf2f55a2
commit 14f2f9328c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/acer/acer_simple.py
View File

@@ -198,7 +198,7 @@ class Model(object):
def save(save_path): def save(save_path):
ps = sess.run(params) ps = sess.run(params)
make_path(save_path) make_path(osp.dirname(save_path))
joblib.dump(ps, save_path) joblib.dump(ps, save_path)
self.train = train self.train = train