From fb63bb543f7e576791321a118daa67417965a9aa Mon Sep 17 00:00:00 2001 From: evonneng Date: Sat, 11 Jun 2022 21:29:16 +0000 Subject: [PATCH] added psnr to the local writer prints of ngp --- pyrad/graphs/instant_ngp.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pyrad/graphs/instant_ngp.py b/pyrad/graphs/instant_ngp.py index 8708d1969..f2fe9b056 100644 --- a/pyrad/graphs/instant_ngp.py +++ b/pyrad/graphs/instant_ngp.py @@ -162,4 +162,6 @@ class NGPGraph(Graph): writer.put_scalar(name=f"ssim/val_{image_idx}", scalar=float(ssim), step=step) writer.put_scalar(name=f"lpips/val_{image_idx}", scalar=float(lpips), step=step) + writer.put_scalar(name=writer.EventName.CURR_TEST_PSNR, scalar=float(psnr), step=step) + return psnr.item()