tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Fix loading in inference mode (#965)

Browse Source
This commit is contained in:
Matthew Tancik 2022-11-15 21:06:54 -08:00 committed by GitHub
parent 95a27ecba5
commit 009e21be24
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nerfstudio/pipelines/base_pipeline.py
View File

@ -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(