Return LR back
@@ -97,7 +97,7 @@ if __name__ == "__main__":
net = Net(obs_size, HIDDEN_SIZE, n_actions)
objective = nn.CrossEntropyLoss()
- optimizer = optim.Adam(params=net.parameters(), lr=0.0005)
+ optimizer = optim.Adam(params=net.parameters(), lr=0.001)
writer = SummaryWriter(comment="-frozenlake-tweaked")
full_batch = []
