From 974b6a9987d34c892e621b894f0db270e90151f6 Mon Sep 17 00:00:00 2001
From: origamiman72 <ahujaabhik@berkeley.edu>
Date: Sun, 24 Jul 2022 22:16:08 -0700
Subject: [PATCH] Reformatted Rays / Sec

---
 pyrad/utils/writer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyrad/utils/writer.py b/pyrad/utils/writer.py
index 6c7f804c0..9f1871f27 100644
--- a/pyrad/utils/writer.py
+++ b/pyrad/utils/writer.py
@@ -34,6 +34,7 @@ from torchtyping import TensorType
 import wandb
 from pyrad.utils.config import LoggingConfig
 from pyrad.utils.decorators import check_main_thread, decorate_all
+from pyrad.utils.misc import human_format
 
 to8b = lambda x: (255 * torch.clamp(x, min=0, max=1)).to(torch.uint8)
 EVENT_WRITERS = []
@@ -49,7 +50,7 @@ class EventName(enum.Enum):
     ITER_TRAIN_TIME = "Train Iter (time)"
     TOTAL_TRAIN_TIME = "Train Total (time)"
     ETA = "ETA (time)"
-    RAYS_PER_SEC = "Rays Per Sec"
+    RAYS_PER_SEC = "Rays / Sec"
     CURR_TEST_PSNR = "Test PSNR"
 
 
@@ -370,6 +371,8 @@ class LocalWriter(Writer):
             if name in self.stats_to_track:
                 if "(time)" in name:
                     v = _format_time(v)
+                elif "Rays" in name:
+                    v = human_format(v)
                 else:
                     v = f"{v:0.4f}"
                 curr_mssg += f"{v:<20} "