diff --git a/nerfstudio/pipelines/base_pipeline.py b/nerfstudio/pipelines/base_pipeline.py index c0927c57a..d5a8b1ec0 100644 --- a/nerfstudio/pipelines/base_pipeline.py +++ b/nerfstudio/pipelines/base_pipeline.py @@ -360,8 +360,11 @@ class VanillaPipeline(Pipeline): """ state = {key.replace("module.", ""): value for key, value in loaded_state.items()} if self.test_mode == "inference": + state.pop("datamanager.train_camera_optimizer.pose_adjustment", None) state.pop("datamanager.train_ray_generator.image_coords", None) + state.pop("datamanager.train_ray_generator.pose_optimizer.pose_adjustment", None) state.pop("datamanager.eval_ray_generator.image_coords", None) + state.pop("datamanager.eval_ray_generator.pose_optimizer.pose_adjustment", None) self.load_state_dict(state) # type: ignore def get_training_callbacks(