brozek/rltorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fixed not having self.

Browse source
This commit is contained in:
Brandon Rozek 2019-02-03 00:51:24 -05:00
parent 03455accc8
commit e62385b574
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
rltorch/agents/DQNAgent.py
View file

@ -19,11 +19,11 @@ class DQNAgent:
state_batch, action_batch, reward_batch, next_state_batch, not_done_batch = M.zip_batch(minibatch)
# Send to their appropriate devices
state_batch = state_batch.to(net.device)
action_batch = action_batch.to(net.device)
reward_batch = reward_batch.to(net.device)
next_state_batch = next_state_batch.to(net.device)
not_done_batch = not_done_batch.to(net.device)
state_batch = state_batch.to(self.net.device)
action_batch = action_batch.to(self.net.device)
reward_batch = reward_batch.to(self.net.device)
next_state_batch = next_state_batch.to(self.net.device)
not_done_batch = not_done_batch.to(self.net.device)
obtained_values = self.net(state_batch).gather(1, action_batch.view(self.config['batch_size'], 1))