mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
added fps as benchmarking stats (#142)
This commit is contained in:
1 parent
d52c11c04c
commit
cc9a438057
2 files changed
+13
-6
No files matched your search
@@ -68,8 +68,9 @@ def main(args):
|
||||
ckpt = _load_best_ckpt(hydra_dir, config.trainer)
|
||||
|
||||
# run evaluation
|
||||
avg_psnr, avg_rays_per_sec = run_inference(config)
|
||||
benchmarks[dataset] = {"avg psnr": avg_psnr, "avg rays/s": avg_rays_per_sec, "checkpoint": ckpt}
|
||||
stats_dict = run_inference(config)
|
||||
stats_dict["checkpoint"] = ckpt
|
||||
benchmarks[dataset] = stats_dict
|
||||
|
||||
# reset hydra config
|
||||
GlobalHydra.instance().clear()
|
||||
|
||||
Reference in new issue
Block a user