Commit 0b28c525 authored by Avisek Naug's avatar Avisek Naug 🎨
Browse files

Update train returns trained agent

parent 071b1333
......@@ -137,7 +137,7 @@ def train_agent(agent, env, steps=30000, dest='agent_weights.h5f'):
# save latest weights
# TODO: Should this be in production?
# agent.save_weights('./rl_results_local/latestweights.h5f')
return store_weights
return store_weights, agent
class PerformanceMetrics(Callback):
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment