Quick fix to viewer OOM issue (#1053)

quick fix to viewer

Author: ethanweber

nerfstudio/cameras/cameras.py

@@ -244,7 +244,7 @@ def _init_get_times(self, times):
             if times.ndim == 0 or times.shape[-1] != 1:
                 times = times.unsqueeze(-1).to(self.device)
         else:
-            raise ValueError(f"times must be a tensor, got {type(times)}")
+            raise ValueError(f"times must be None or a tensor, got {type(times)}")
 
         return times
 

nerfstudio/viewer/server/viewer_utils.py

@@ -325,7 +325,7 @@ def init_scene(self, dataset: InputDataset, start_train=True) -> None:
         # set the initial state whether to train or not
         self.vis["renderingState/isTraining"].write(start_train)
 
-        self.vis["renderingState/render_time"].write(str(0))
+        # self.vis["renderingState/render_time"].write(str(0))
 
         # set the properties of the camera
         # self.vis["renderingState/camera"].write(json_)
@@ -756,8 +756,9 @@ def _render_image_in_viewer(self, camera_object, graph: Model, is_training: bool
             dim=0,
         )
 
-        times = float(self.vis["renderingState/render_time"].read())
-        times = torch.tensor([times])
+        times = self.vis["renderingState/render_time"].read()
+        if times:
+            times = torch.tensor([float(times)])
 
         camera = Cameras(
             fx=intrinsics_matrix[0, 0],