File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
ml-agents/mlagents/trainers/torch Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -569,7 +569,7 @@ def critic_pass(
569
569
if self .use_lstm :
570
570
# Use only the back half of memories for critic
571
571
actor_mem , critic_mem = torch .split (memories , self .memory_size // 2 , - 1 )
572
- all_net_inputs = [net_inputs ]
572
+ all_net_inputs = []
573
573
if critic_obs is not None :
574
574
all_net_inputs .extend (critic_obs )
575
575
value_outputs , critic_mem_out = self .critic (
@@ -601,7 +601,7 @@ def get_stats_and_value(
601
601
net_inputs , memories = actor_mem , sequence_length = sequence_length
602
602
)
603
603
log_probs , entropies = self .action_model .evaluate (encoding , masks , actions )
604
- all_net_inputs = [net_inputs ]
604
+ all_net_inputs = []
605
605
if critic_obs is not None :
606
606
all_net_inputs .extend (critic_obs )
607
607
value_outputs , critic_mem_outs = self .critic (
@@ -627,7 +627,7 @@ def get_action_stats_and_value(
627
627
critic_mem = None
628
628
actor_mem = None
629
629
630
- all_net_inputs = [net_inputs ]
630
+ all_net_inputs = []
631
631
if critic_obs is not None :
632
632
all_net_inputs .extend (critic_obs )
633
633
You can’t perform that action at this time.
0 commit comments