Fix splatfacto crash in eval when using viewer (nerfstudio-project#3430)

jb-ye · Sep 17, 2024 · 516fd7c · 516fd7c
1 parent 3e638da
commit 516fd7c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py
@@ -300,7 +300,8 @@ def train(self) -> None:
 
                 # Do not perform evaluation if there are no validation images
                 if self.pipeline.datamanager.eval_dataset:
-                    self.eval_iteration(step)
+                    with self.train_lock:
+                        self.eval_iteration(step)
 
                 if step_check(step, self.config.steps_per_save):
                     self.save_checkpoint(step)