From 24ab477b4541fefce00c73385302f21066f8e0b0 Mon Sep 17 00:00:00 2001 From: evonneng Date: Mon, 25 Jul 2022 21:22:29 +0000 Subject: [PATCH] fixed dataset loading in benchmark --- scripts/benchmarking/run_benchmark.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/benchmarking/run_benchmark.py b/scripts/benchmarking/run_benchmark.py index b0c31b9d3..38a2431e4 100644 --- a/scripts/benchmarking/run_benchmark.py +++ b/scripts/benchmarking/run_benchmark.py @@ -41,7 +41,6 @@ def _load_best_ckpt(hydra_dir: str, config: DictConfig) -> str: step = os.path.splitext(latest_ckpt)[0].split("-")[-1] config.resume_train.load_dir = model_dir config.resume_train.load_step = int(step) - config.data.dataloader_eval.image_indices = None return os.path.join(model_dir, latest_ckpt) @@ -76,6 +75,8 @@ def main(args): # set up trainer, config, and checkpoint loading hydra_dir = f"{args.hydra_base_dir}/blender_{dataset}_{args.benchmark_date}/{args.graph}/" config = _load_hydra_config(hydra_dir) + config.data.dataloader_eval.image_indices = None + ckpt = _load_best_ckpt(hydra_dir, config.trainer) # run evaluation