gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-03-13 04:55:47 +01:00
Code Issues Projects Releases Wiki Activity

Re-enable NFS data store. (#101)

Browse Source
This commit is contained in:
Balaji Subramaniam
2018-11-16 13:55:33 -08:00
committed by GitHub
parent a0f25034c3
commit dea1826658
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/rollout_worker.py
View File

@@ -68,6 +68,8 @@ def get_latest_checkpoint(checkpoint_dir):
rel_path = os.path.relpath(ckpt.model_checkpoint_path, checkpoint_dir)
return int(rel_path.split('_Step')[0])
return 0
def should_stop(checkpoint_dir):
return os.path.exists(os.path.join(checkpoint_dir, SyncFiles.FINISHED.value))