From a73d892f21ce3be4050a7e4dc97f4a801e0b883a Mon Sep 17 00:00:00 2001 From: Justin Date: Wed, 21 Jun 2023 21:02:07 +0000 Subject: [PATCH] params --- nerfstudio/data/datamanagers/parallel_datamanager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/data/datamanagers/parallel_datamanager.py b/nerfstudio/data/datamanagers/parallel_datamanager.py index 60affa17b..57fdac1ca 100644 --- a/nerfstudio/data/datamanagers/parallel_datamanager.py +++ b/nerfstudio/data/datamanagers/parallel_datamanager.py @@ -109,7 +109,7 @@ class DataProc(mp.Process): self.out_queue.put_nowait((ray_bundle, batch)) break except: - time.sleep(0.005) + time.sleep(0.001) def cache_images(self): # caches all the input images in a NxHxWx3 tensor @@ -160,7 +160,7 @@ class ParallelDataManager(DataManager, Generic[TDataset]): mp.set_start_method("spawn") # Manager().queue() can be faster # https://stackoverflow.com/questions/43439194/python-multiprocessing-queue-vs-multiprocessing-manager-queue/45236748#45236748 - self.data_q = mp.Manager().Queue(maxsize=4) + self.data_q = mp.Manager().Queue(maxsize=2) self.data_procs = [ DataProc(self.data_q, self.config, self.train_dataparser_outputs, self.train_dataset, self.pix_sampler) for i in range(self.config.n_procs)