gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-14 02:22:31 +02:00
Code Issues Projects Releases Wiki Activity

rename save_checkpoint_dir -> checkpoint_save_dir

Browse Source
This commit is contained in:
Zach Dwiel authored and zach dwiel committed 2018-10-23 17:10:58 -04:00
1 parent 201a2237a1
commit 9804b033a2
4 files changed
+5 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

rl_coach/agents/nec_agent.py
+1 -1
View File
@@ -171,5 +171,5 @@ class NECAgent(ValueOptimizationAgent):
actions, returns)
def save_checkpoint(self, checkpoint_id):
with open(os.path.join(self.ap.task_parameters.save_checkpoint_dir, str(checkpoint_id) + '.dnd'), 'wb') as f:
with open(os.path.join(self.ap.task_parameters.checkpoint_save_dir, str(checkpoint_id) + '.dnd'), 'wb') as f:
pickle.dump(self.networks['main'].online_network.output_heads[0].DND, f, pickle.HIGHEST_PROTOCOL)