From 9c96374139351e5a1d8dc2f95937cff97049c17b Mon Sep 17 00:00:00 2001
From: Rong Liu <102014841+RongLiu-Leo@users.noreply.github.com>
Date: Thu, 1 May 2025 16:31:53 -0700
Subject: [PATCH] avoid a division by zero

---
 examples/01_dummy_training.py  | 2 +-
 examples/04_gsplat_training.py | 4 ++--
 nerfview/_renderer.py          | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/01_dummy_training.py b/examples/01_dummy_training.py
index 9d8d51a..5bf13d7 100644
--- a/examples/01_dummy_training.py
+++ b/examples/01_dummy_training.py
@@ -100,7 +100,7 @@ def training_step():
         tic = time.time()
         with viewer.lock:
             num_train_rays_per_step = training_step()
-        num_train_steps_per_sec = 1.0 / (time.time() - tic)
+        num_train_steps_per_sec = 1.0 / (max(time.time() - tic, 1e-10))
         num_train_rays_per_sec = num_train_rays_per_step * num_train_steps_per_sec
         # Update the viewer state.
         viewer.render_tab_state.num_train_rays_per_sec = num_train_rays_per_sec
diff --git a/examples/04_gsplat_training.py b/examples/04_gsplat_training.py
index cf8debe..85cb0de 100644
--- a/examples/04_gsplat_training.py
+++ b/examples/04_gsplat_training.py
@@ -631,7 +631,7 @@ def train(self):
 
             if not cfg.disable_viewer:
                 self.viewer.lock.release()
-                num_train_steps_per_sec = 1.0 / (time.time() - tic)
+                num_train_steps_per_sec = 1.0 / (max(time.time() - tic, 1e-10))
                 num_train_rays_per_sec = (
                     num_train_rays_per_step * num_train_steps_per_sec
                 )
@@ -822,7 +822,7 @@ def eval(self, step: int):
             )  # [1, H, W, 3]
             colors = torch.clamp(colors, 0.0, 1.0)
             torch.cuda.synchronize()
-            ellipse_time += time.time() - tic
+            ellipse_time += max(time.time() - tic, 1e-10)
 
             # write images
             canvas = torch.cat([pixels, colors], dim=2).squeeze(0).cpu().numpy()
diff --git a/nerfview/_renderer.py b/nerfview/_renderer.py
index 0d55982..ad17ca3 100644
--- a/nerfview/_renderer.py
+++ b/nerfview/_renderer.py
@@ -171,7 +171,7 @@ def run(self):
                     else:
                         img, depth = rendered, None
                     self.viewer.render_tab_state.num_view_rays_per_sec = (W * H) / (
-                        time.time() - tic
+                        max(time.time() - tic, 1e-10)
                     )
             except InterruptRenderException:
                 continue