Add comment to ghost trainer

Unity-Technologies · Dec 18, 2020 · 292b6ce · 292b6ce
1 parent afd7476
commit 292b6ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ml-agents/mlagents/trainers/torch/networks.py b/ml-agents/mlagents/trainers/torch/networks.py
@@ -569,7 +569,7 @@ def critic_pass(
         if self.use_lstm:
             # Use only the back half of memories for critic
             actor_mem, critic_mem = torch.split(memories, self.memory_size // 2, -1)
-        all_net_inputs = [net_inputs]
+        all_net_inputs = []
         if critic_obs is not None:
             all_net_inputs.extend(critic_obs)
         value_outputs, critic_mem_out = self.critic(
@@ -601,7 +601,7 @@ def get_stats_and_value(
             net_inputs, memories=actor_mem, sequence_length=sequence_length
         )
         log_probs, entropies = self.action_model.evaluate(encoding, masks, actions)
-        all_net_inputs = [net_inputs]
+        all_net_inputs = []
         if critic_obs is not None:
             all_net_inputs.extend(critic_obs)
         value_outputs, critic_mem_outs = self.critic(
@@ -627,7 +627,7 @@ def get_action_stats_and_value(
             critic_mem = None
             actor_mem = None
 
-        all_net_inputs = [net_inputs]
+        all_net_inputs = []
         if critic_obs is not None:
             all_net_inputs.extend(critic_obs)