Commit 9f0e8ab6 authored by lli's avatar lli
Browse files

corrected bugs

parent bb7d426f
......@@ -59,7 +59,7 @@ print(f'Discount factor: {gamma}')
print()
# Initialize the policy network
policy_net = PolicyNetwork(n_state, n_action, args.n_hidden, args.lr_policy)
policy_net = PolicyNetwork(n_state, args.n_hidden, n_action, args.lr_policy)
# Initialize the value network
value_net = ValueNetwork(n_state, args.n_hidden, args.lr_value)
......
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