Skip to content

Commit

Permalink
Update separated_buffer_macpo.py
Browse files Browse the repository at this point in the history
  • Loading branch information
chauncygu authored Sep 12, 2022
1 parent 6c4285a commit 33d747c
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -309,7 +309,7 @@ def feed_forward_generator(self, advantages, num_mini_batch=None, mini_batch_siz
if self.algo == "macppo":
factor_batch = factor[indices]
yield share_obs_batch, obs_batch, rnn_states_batch, rnn_states_critic_batch, actions_batch, value_preds_batch, return_batch, masks_batch, active_masks_batch, old_action_log_probs_batch, adv_targ, available_actions_batch, factor_batch, cost_preds_batch, cost_return_batch, rnn_states_cost_batch, cost_adv_targ
elif self.algo == "happolag":
elif self.algo == "mappolag":
factor_batch = factor[indices]
yield share_obs_batch, obs_batch, rnn_states_batch, rnn_states_critic_batch, actions_batch, value_preds_batch, return_batch, masks_batch, active_masks_batch, old_action_log_probs_batch, adv_targ, available_actions_batch, factor_batch, cost_preds_batch, cost_return_batch, rnn_states_cost_batch, cost_adv_targ
elif self.algo == "macpo":
Expand Down

0 comments on commit 33d747c

Please sign in to comment.