finally block

nerfstudio-project · Oct 18, 2024 · b32e793 · b32e793
1 parent 5dae507
commit b32e793
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/nerfstudio/viewer/render_state_machine.py b/nerfstudio/viewer/render_state_machine.py
@@ -169,11 +169,10 @@ def _render_img(self, camera_state: CameraState):
                         with torch.no_grad(), viewer_utils.SetTrace(self.check_interrupt):
                             outputs = self.viewer.get_model().get_outputs_for_camera(camera, obb_box=obb)
                 except viewer_utils.IOChangeException:
+                    raise
+                finally:
                     if was_training:
                         self.viewer.get_model().train()
-                    raise
-                if was_training:
-                    self.viewer.get_model().train()
             num_rays = (camera.height * camera.width).item()
             if self.viewer.control_panel.layer_depth:
                 if isinstance(self.viewer.get_model(), SplatfactoModel):