Commit 8738c2b3 authored by lli's avatar lli
Browse files

corrected the number of hidden layers in dqn

parent 10901380
......@@ -22,6 +22,8 @@ class DQN(nn.Module):
torch.nn.ReLU(),
torch.nn.Linear(n_hidden, n_hidden),
torch.nn.ReLU(),
torch.nn.Linear(n_hidden, n_hidden),
torch.nn.ReLU(),
torch.nn.Linear(n_hidden, n_action)
)
......
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