Corrections
This commit is contained in:
parent
5ad1cf33b5
commit
634840accb
|
@ -904,7 +904,7 @@ class SemanticKittiSampler(Sampler):
|
||||||
for batch_i, batch in enumerate(dataloader):
|
for batch_i, batch in enumerate(dataloader):
|
||||||
|
|
||||||
# Control max_in_points value
|
# Control max_in_points value
|
||||||
all_lengths += batch.lengths[0]
|
all_lengths += batch.lengths[0].tolist()
|
||||||
|
|
||||||
# Convergence
|
# Convergence
|
||||||
if len(all_lengths) > N:
|
if len(all_lengths) > N:
|
||||||
|
|
Loading…
Reference in a new issue