From 497cc2e9fdd3cf2d56454db7f6986fc2f5eaddcd Mon Sep 17 00:00:00 2001 From: jdsr <107491647+jdsr1145@users.noreply.github.com> Date: Mon, 19 Jan 2026 16:51:01 -0800 Subject: [PATCH] Change data source for next_eval to eval dataloader Implementation of issue #3731 --- nerfstudio/data/datamanagers/parallel_datamanager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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