Skip to content

Commit 02e35fd

Browse files
author
Ervin T
authored
[bug-fix] Fix error with discrete probs (#4309)
1 parent 949aa1f commit 02e35fd

File tree

1 file changed

+1
-1
lines changed
  • ml-agents/mlagents/trainers/torch

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,5 +162,5 @@ def get_probs_and_entropy(
162162
entropies = entropies.squeeze(-1)
163163
all_probs = None
164164
else:
165-
all_probs = torch.cat(all_probs, dim=-1)
165+
all_probs = torch.cat(all_probs_list, dim=-1)
166166
return log_probs, entropies, all_probs

0 commit comments

Comments
 (0)