failed attempt correction
This commit is contained in:
parent
5b5641e02d
commit
7255680ff0
|
@ -344,7 +344,7 @@ class S3DISDataset(PointCloudDataset):
|
|||
# Safe check for empty spheres
|
||||
if n < 2:
|
||||
failed_attempts += 1
|
||||
if failed_attempts > 2 * self.config.batch_num:
|
||||
if failed_attempts > 100 * self.config.batch_num:
|
||||
raise ValueError('It seems this dataset only containes empty input spheres')
|
||||
t += [time.time()]
|
||||
t += [time.time()]
|
||||
|
@ -550,7 +550,7 @@ class S3DISDataset(PointCloudDataset):
|
|||
# Safe check for empty spheres
|
||||
if n < 2:
|
||||
failed_attempts += 1
|
||||
if failed_attempts > 2 * self.config.batch_num:
|
||||
if failed_attempts > 100 * self.config.batch_num:
|
||||
raise ValueError('It seems this dataset only containes empty input spheres')
|
||||
continue
|
||||
|
||||
|
|
Loading…
Reference in a new issue