diff --git a/nerfstudio/data/datamanagers/parallel_datamanager.py b/nerfstudio/data/datamanagers/parallel_datamanager.py index e1e1f6ef52..5dda767a37 100644 --- a/nerfstudio/data/datamanagers/parallel_datamanager.py +++ b/nerfstudio/data/datamanagers/parallel_datamanager.py @@ -248,7 +248,7 @@ def next_train(self, step: int) -> Tuple[RayBundle, Dict]: def next_eval(self, step: int) -> Tuple[RayBundle, Dict]: """Returns the next batch of data from the eval dataloader.""" self.eval_count += 1 - ray_bundle, batch = next(self.iter_train_raybundles)[0] + ray_bundle, batch = next(self.iter_eval_raybundles)[0] ray_bundle = ray_bundle.to(self.device) batch = get_dict_to_torch(batch, self.device) return ray_bundle, batch