failed attempt correction

HuguesTHOMAS · Apr 6, 2022 · 7255680 · 7255680
1 parent 5b5641e
commit 7255680
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets/S3DIS.py b/datasets/S3DIS.py
@@ -344,7 +344,7 @@ def potential_item(self, batch_i, debug_workers=False):
             # 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 @@ def random_item(self, batch_i):
             # 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