From 797e9012fb8bdb167cdf7f31ead595d864ada2a9 Mon Sep 17 00:00:00 2001 From: Dylan Hu Date: Tue, 18 Apr 2023 02:33:54 -0400 Subject: [PATCH] Fix divide by zero when train util set to 1 (#1757) * fix divide by zero when train util set to 1 * Run black --------- Co-authored-by: Brent Yi --- nerfstudio/viewer/server/viewer_state.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/viewer/server/viewer_state.py b/nerfstudio/viewer/server/viewer_state.py index beee00ba7..54660b666 100644 --- a/nerfstudio/viewer/server/viewer_state.py +++ b/nerfstudio/viewer/server/viewer_state.py @@ -303,7 +303,7 @@ class ViewerState: if self.camera_message is None: return - if self.trainer is not None and self.trainer.is_training: + if self.trainer is not None and self.trainer.is_training and self.control_panel.train_util != 1: if ( EventName.TRAIN_RAYS_PER_SEC.value in GLOBAL_BUFFER["events"] and EventName.VIS_RAYS_PER_SEC.value in GLOBAL_BUFFER["events"]