Skip to content

Commit 292b6ce

Browse files
author
Ervin Teng
committed
Add comment to ghost trainer
1 parent afd7476 commit 292b6ce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ml-agents/mlagents/trainers/torch/networks.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -569,7 +569,7 @@ def critic_pass(
569569
if self.use_lstm:
570570
# Use only the back half of memories for critic
571571
actor_mem, critic_mem = torch.split(memories, self.memory_size // 2, -1)
572-
all_net_inputs = [net_inputs]
572+
all_net_inputs = []
573573
if critic_obs is not None:
574574
all_net_inputs.extend(critic_obs)
575575
value_outputs, critic_mem_out = self.critic(
@@ -601,7 +601,7 @@ def get_stats_and_value(
601601
net_inputs, memories=actor_mem, sequence_length=sequence_length
602602
)
603603
log_probs, entropies = self.action_model.evaluate(encoding, masks, actions)
604-
all_net_inputs = [net_inputs]
604+
all_net_inputs = []
605605
if critic_obs is not None:
606606
all_net_inputs.extend(critic_obs)
607607
value_outputs, critic_mem_outs = self.critic(
@@ -627,7 +627,7 @@ def get_action_stats_and_value(
627627
critic_mem = None
628628
actor_mem = None
629629

630-
all_net_inputs = [net_inputs]
630+
all_net_inputs = []
631631
if critic_obs is not None:
632632
all_net_inputs.extend(critic_obs)
633633

0 commit comments

Comments
 (0)