Commit 1059735f authored by lli's avatar lli
Browse files

add training time to a2c

parent decea86e
......@@ -135,6 +135,8 @@ for episode in range(n_episode):
state = next_state
print(f"Training time for {args.n_episode} episodes: {timer() - start_time}")
# save the model parameters
torch.save(policy_net.state_dict(), os.path.join(OUT_PATH, 'policy_net{}.pk1'.format(n_episode)))
......
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