Add disable_video flag for simple_trainer

nerfstudio-project · Feb 13, 2025 · c95fcff · c95fcff
1 parent 0880d2b
commit c95fcff
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/simple_trainer.py b/examples/simple_trainer.py
@@ -90,6 +90,8 @@ class Config:
     save_ply: bool = False
     # Steps to save the model as ply
     ply_steps: List[int] = field(default_factory=lambda: [7_000, 30_000])
+    # Whether to disable video generation during training and evaluation
+    disable_video: bool = False
 
     # Initialization strategy
     init_type: str = "sfm"
@@ -932,6 +934,8 @@ def eval(self, step: int, stage: str = "val"):
     @torch.no_grad()
     def render_traj(self, step: int):
         """Entry for trajectory rendering."""
+        if self.cfg.disable_video:
+            return
         print("Running trajectory rendering...")
         cfg = self.cfg
         device = self.device