added fps as benchmarking stats (#142)

This commit is contained in:
evonneng authored and GitHub committed 2022-07-14 17:04:47 -07:00
1 parent d52c11c04c
commit cc9a438057
2 files changed
+13 -6

No files matched your search

+3 -2
View File
@@ -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()