Skip to content

Commit 516fd7c

Browse files
authored
Fix splatfacto crash in eval when using viewer (#3430)
1 parent 3e638da commit 516fd7c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nerfstudio/engine/trainer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,8 @@ def train(self) -> None:
300300

301301
# Do not perform evaluation if there are no validation images
302302
if self.pipeline.datamanager.eval_dataset:
303-
self.eval_iteration(step)
303+
with self.train_lock:
304+
self.eval_iteration(step)
304305

305306
if step_check(step, self.config.steps_per_save):
306307
self.save_checkpoint(step)

0 commit comments

Comments
 (0)