From 79b3e134fed0019bed135dbd5bf7f5d1316209db Mon Sep 17 00:00:00 2001 From: Ervin Teng Date: Wed, 5 Aug 2020 17:03:17 -0700 Subject: [PATCH] Fix error with discrete probs --- ml-agents/mlagents/trainers/torch/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml-agents/mlagents/trainers/torch/utils.py b/ml-agents/mlagents/trainers/torch/utils.py index 600c7eb8d9..c459c9ef79 100644 --- a/ml-agents/mlagents/trainers/torch/utils.py +++ b/ml-agents/mlagents/trainers/torch/utils.py @@ -162,5 +162,5 @@ def get_probs_and_entropy( entropies = entropies.squeeze(-1) all_probs = None else: - all_probs = torch.cat(all_probs, dim=-1) + all_probs = torch.cat(all_probs_list, dim=-1) return log_probs, entropies, all_probs