diff --git a/scripts/benchmarking/run_benchmark.py b/scripts/benchmarking/run_benchmark.py index 38a2431e4..2a9970c18 100644 --- a/scripts/benchmarking/run_benchmark.py +++ b/scripts/benchmarking/run_benchmark.py @@ -76,7 +76,7 @@ def main(args): 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