Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Corrections

Browse source
This commit is contained in:
HuguesTHOMAS 2020-05-07 23:32:00 -04:00
parent 5ad1cf33b5
commit 634840accb
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
datasets/SemanticKitti.py
View file

@ -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: