Datamanager eval fix (#1482)

* Fix datamanager

* Datamanager tweak

* Hopefully fixed now

* readd comment
This commit is contained in:
Matthew Tancik authored and GitHub committed 2023-02-22 16:53:23 -08:00
1 parent 50dc5d6cd7
commit 69fdc7db9c
1 file changed
+4 -1
@@ -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(