Skip to content

Commit 2ed7c69

Browse files
authored
Updated model
1 parent 32da52a commit 2ed7c69

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dqn_agent.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ def __init__(self, state_size, action_size):
3030

3131
self.local_nn = Model(state_size, action_size).to(device)
3232
# self.target_nn = Model(state_size, action_size).to(device)
33-
self.qnetwork_local.load_state_dict(torch.load('/Users/sanketsans/Documents/Udacity/deepRL/deep-reinforcement-learning/p1_navigation/checkpoint0.pth', map_location=torch.device('cpu')))
34-
self.qnetwork_local.eval()
33+
self.local_nn.load_state_dict(torch.load('/Users/sanketsans/Documents/Udacity/deepRL/deep-reinforcement-learning/p1_navigation/checkpoint0.pth', map_location=torch.device('cpu')))
34+
self.local_nn.eval()
3535

3636
self.optimizer = optim.Adam(self.local_nn.parameters(), lr=LR)
3737

0 commit comments

Comments
 (0)