gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

rename save_checkpoint_dir -> checkpoint_save_dir

Browse Source
This commit is contained in:
Zach Dwiel
2018-10-05 11:44:49 -04:00
committed by zach dwiel
parent 201a2237a1
commit 9804b033a2
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/agents/nec_agent.py
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)