diff --git a/nerfstudio/data/datamanagers/parallel_datamanager.py b/nerfstudio/data/datamanagers/parallel_datamanager.py index e7521beca..b28e530f9 100644 --- a/nerfstudio/data/datamanagers/parallel_datamanager.py +++ b/nerfstudio/data/datamanagers/parallel_datamanager.py @@ -93,10 +93,10 @@ class DataProcessor(mp.Process): # type: ignore self.exclude_batch_keys_from_device = self.dataset.exclude_batch_keys_from_device self.pixel_sampler = pixel_sampler self.ray_generator = RayGenerator(self.dataset.cameras) - self.cache_images() def run(self): """Append out queue in parallel with ray bundles and batches.""" + self.cache_images() while True: batch = self.pixel_sampler.sample(self.img_data) ray_indices = batch["indices"]