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

Non-strict loading of checkpoints (#1235)

Browse Source
This commit is contained in:
Matthew Tancik 2023-01-16 15:12:59 -08:00 committed by GitHub
parent f0b4f5b35e
commit 23ca7a7cc8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nerfstudio/pipelines/base_pipeline.py
View File

@ -359,13 +359,7 @@ class VanillaPipeline(Pipeline):
loaded_state: pre-trained model state dict
"""
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
self.load_state_dict(state, strict=False)
def get_training_callbacks(
self, training_callback_attributes: TrainingCallbackAttributes