public
/
ottertune
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bug fix: pass single result object to save_execution_time()

This commit is contained in:
dvanaken 2020-04-03 02:30:04 -04:00 committed by Dana Van Aken
parent 94dcc0530d
commit 6d123d7509
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/website/website/tasks/async_tasks.py
View File

@ -624,7 +624,7 @@ def train_ddpg(train_ddpg_input):
session.ddpg_actor_model, session.ddpg_critic_model = ddpg.get_model() session.ddpg_actor_model, session.ddpg_critic_model = ddpg.get_model()
session.ddpg_reply_memory = ddpg.replay_memory.get() session.ddpg_reply_memory = ddpg.replay_memory.get()
session.save() session.save()
exec_time = save_execution_time(start_ts, "train_ddpg", result) exec_time = save_execution_time(start_ts, "train_ddpg", result.first())
LOG.debug("\n%s: Result = %s\n", task_name, _task_result_tostring(target_data)) LOG.debug("\n%s: Result = %s\n", task_name, _task_result_tostring(target_data))
LOG.info('%s: Done training ddpg (%.1f seconds).', task_name, exec_time) LOG.info('%s: Done training ddpg (%.1f seconds).', task_name, exec_time)
return target_data, algorithm return target_data, algorithm