wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-08-16 11:26:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Pranjal Tandon
2018-09-21 00:32:20 +05:30
committed by GitHub
parent 02865da6b1
commit ea6881895b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+3 -1
View File
@@ -75,7 +75,9 @@ for i_episode in itertools.count():
# Sample a batch from memory
state_batch, action_batch, reward_batch, next_state_batch, mask_batch = memory.sample(args.batch_size)
# Update parameters of all the networks
agent.update_parameters(state_batch, action_batch, reward_batch, next_state_batch, mask_batch, updates)
value_loss, critic_1_loss, critic_2_loss, policy_loss = agent.update_parameters(state_batch, action_batch,
reward_batch, next_state_batch,
mask_batch, updates)
writer.add_scalar('loss/value', value_loss, updates)
writer.add_scalar('loss/critic_1', critic_1_loss, updates)
writer.add_scalar('loss/critic_2', critic_2_loss, updates)