From 516fd7c9ac73e28db6a522df57d7b22e7b0d6756 Mon Sep 17 00:00:00 2001
From: Aayush Gupta <19579293+aayushg55@users.noreply.github.com>
Date: Tue, 17 Sep 2024 12:31:42 -0700
Subject: [PATCH] Fix splatfacto crash in eval when using viewer (#3430)

---
 nerfstudio/engine/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py
index 644d62a5b5..cac31cf504 100644
--- 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)