From 6d898272373240ad4e2fa2ca64d67d3e6cbac8c4 Mon Sep 17 00:00:00 2001 From: haoshengzou Date: Thu, 11 Mar 2021 16:03:07 +0800 Subject: [PATCH] minor fix in comments of 'get_path_indices()' --- src/policy_gradients/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/policy_gradients/torch_utils.py b/src/policy_gradients/torch_utils.py index f8590a5..5a97018 100644 --- a/src/policy_gradients/torch_utils.py +++ b/src/policy_gradients/torch_utils.py @@ -208,7 +208,7 @@ def get_path_indices(not_dones): tensor([[1, 1, 0, 1, 1, 1, 1, 1, 1, 1], [1, 1, 0, 1, 1, 0, 1, 1, 0, 1]], dtype=torch.uint8) Then we would return: - [(0, 0, 3), (0, 3, 10), (1, 0, 3), (1, 3, 5), (1, 5, 9), (1, 9, 10)] + [(0, 0, 3), (0, 3, 10), (1, 0, 3), (1, 3, 6), (1, 6, 9), (1, 9, 10)] """ indices = [] num_timesteps = not_dones.shape[1]