From 634840accb28b1731fe0c4cac4547dd64c4d43bd Mon Sep 17 00:00:00 2001 From: HuguesTHOMAS Date: Thu, 7 May 2020 23:32:00 -0400 Subject: [PATCH] Corrections --- datasets/SemanticKitti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/SemanticKitti.py b/datasets/SemanticKitti.py index 215d931..a7d2ab9 100644 --- a/datasets/SemanticKitti.py +++ b/datasets/SemanticKitti.py @@ -904,7 +904,7 @@ class SemanticKittiSampler(Sampler): for batch_i, batch in enumerate(dataloader): # Control max_in_points value - all_lengths += batch.lengths[0] + all_lengths += batch.lengths[0].tolist() # Convergence if len(all_lengths) > N: