From b046e3b9751c96bc393db91b49301aced98cb2b9 Mon Sep 17 00:00:00 2001 From: Justin Kerr Date: Wed, 16 Oct 2024 12:31:49 -0700 Subject: [PATCH] add ability to load config.yml from a folder not the same as the original training one --- nerfstudio/utils/eval_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nerfstudio/utils/eval_utils.py b/nerfstudio/utils/eval_utils.py index 11a8b2341..3401b80db 100644 --- a/nerfstudio/utils/eval_utils.py +++ b/nerfstudio/utils/eval_utils.py @@ -99,7 +99,8 @@ def eval_setup( # load checkpoints from wherever they were saved # TODO: expose the ability to choose an arbitrary checkpoint - config.load_dir = config.get_checkpoint_dir() + # By using the parent of config_path, we can call this from any folder even if it wasn't originally trained from there + config.load_dir = config_path.parent / config.relative_model_dir # setup pipeline (which includes the DataManager) device = torch.device("cuda" if torch.cuda.is_available() else "cpu")