From 69fdc7db9c700296557c885305b571fec20144c0 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Wed, 22 Feb 2023 16:53:23 -0800 Subject: [PATCH] Datamanager eval fix (#1482) * Fix datamanager * Datamanager tweak * Hopefully fixed now * readd comment --- nerfstudio/data/datamanagers/base_datamanager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index e48218c39..f50fc0361 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -402,9 +402,12 @@ class VanillaDataManager(DataManager): # pylint: disable=abstract-method ) self.iter_eval_image_dataloader = iter(self.eval_image_dataloader) self.eval_pixel_sampler = self._get_pixel_sampler(self.eval_dataset, self.config.eval_num_rays_per_batch) + self.eval_camera_optimizer = self.config.camera_optimizer.setup( + num_cameras=self.eval_dataset.cameras.size, device=self.device + ) self.eval_ray_generator = RayGenerator( self.eval_dataset.cameras.to(self.device), - self.train_camera_optimizer, # should be shared between train and eval. + self.eval_camera_optimizer, ) # for loading full images self.fixed_indices_eval_dataloader = FixedIndicesEvalDataloader(