From 009e21be248418028e6303ecde593a20ef88fa87 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Tue, 15 Nov 2022 21:06:54 -0800 Subject: [PATCH] Fix loading in inference mode (#965) --- nerfstudio/pipelines/base_pipeline.py | 3 +++ 1 file changed, 3 insertions(+) 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(