Commit 6fa38e3c authored by Avisek Naug's avatar Avisek Naug 🎨
Browse files

Update agent.py: remove save weights

parent 2b6e814c
......@@ -136,7 +136,7 @@ def train_agent(agent, env, steps=30000, dest='agent_weights.h5f'):
agent.fit(env, nb_steps=steps, visualize=False, verbose=1, callbacks=[store_weights])
# save latest weights
# TODO: Should this be in production?
agent.save_weights('./rl_results_local/latestweights.h5f')
# agent.save_weights('./rl_results_local/latestweights.h5f')
return store_weights
......
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