From 3b8e2ce9ec4750b21b6b673bf62bedb61e1da2cf Mon Sep 17 00:00:00 2001 From: Mason McGough Date: Fri, 20 Jan 2023 19:11:10 -0500 Subject: [PATCH] Add `step` argument in `eval_utils.py` to fix rendering bug. (#1272) --- nerfstudio/utils/eval_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/utils/eval_utils.py b/nerfstudio/utils/eval_utils.py index 17b0d4b06..3431b20a3 100644 --- a/nerfstudio/utils/eval_utils.py +++ b/nerfstudio/utils/eval_utils.py @@ -59,7 +59,7 @@ def eval_load_checkpoint(config: TrainerConfig, pipeline: Pipeline) -> Path: load_path = config.load_dir / f"step-{load_step:09d}.ckpt" assert load_path.exists(), f"Checkpoint {load_path} does not exist" loaded_state = torch.load(load_path, map_location="cpu") - pipeline.load_pipeline(loaded_state["pipeline"]) + pipeline.load_pipeline(loaded_state["pipeline"], loaded_state["step"]) CONSOLE.print(f":white_check_mark: Done loading checkpoint from {load_path}") return load_path