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(