Skip to content

Commit

Permalink
Update rl_trainer.py
Browse files Browse the repository at this point in the history
  • Loading branch information
mdeib authored Aug 3, 2020
1 parent c720f76 commit 5a20c55
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions hw2/cs285/infrastructure/rl_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,8 +147,9 @@ def train_agent(self):
print('\nTraining agent using sampled data from replay buffer...')
for train_step in range(self.params['num_agent_train_steps_per_iter']):
ob_batch, ac_batch, re_batch, next_ob_batch, terminal_batch = self.agent.sample(self.params['train_batch_size'])

return self.agent.train(ob_batch, ac_batch, re_batch, next_ob_batch, terminal_batch)
loss = self.agent.train(ob_batch, ac_batch, re_batch, next_ob_batch, terminal_batch)

return loss

####################################
####################################
Expand Down

0 comments on commit 5a20c55

Please sign in to comment.