diff --git a/datasets/S3DIS.py b/datasets/S3DIS.py index e183d7d..0b4054e 100644 --- a/datasets/S3DIS.py +++ b/datasets/S3DIS.py @@ -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