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

Update remove save weights

parent 2b6e814c
......@@ -136,7 +136,7 @@ def train_agent(agent, env, steps=30000, dest='agent_weights.h5f'):, 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')
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