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

logging screen output to file (#479)

Browse Source 
Co-authored-by: Gal Leibovich <gal.leibovich@intel.com>
This commit is contained in:
Guy Jacob
2021-05-06 18:02:27 +03:00
committed by GitHub
parent 9106b69227
commit a1a2e67fbd
8 changed files with 63 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rl_coach/coach.py
View File

@@ -174,19 +174,19 @@ def handle_distributed_coach_orchestrator(args):
data_store_params=ds_params_instance)
orchestrator = Kubernetes(orchestration_params)
if not orchestrator.setup(args.checkpoint_restore_dir):
print("Could not setup.")
screen.print("Could not setup.")
return 1
if orchestrator.deploy_trainer():
print("Successfully deployed trainer.")
screen.print("Successfully deployed trainer.")
else:
print("Could not deploy trainer.")
screen.print("Could not deploy trainer.")
return 1
if orchestrator.deploy_worker():
print("Successfully deployed rollout worker(s).")
screen.print("Successfully deployed rollout worker(s).")
else:
print("Could not deploy rollout worker(s).")
screen.print("Could not deploy rollout worker(s).")
return 1
if args.dump_worker_logs: