diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py index 0ce05db0b..0e194b631 100644 --- a/nerfstudio/engine/trainer.py +++ b/nerfstudio/engine/trainer.py @@ -266,7 +266,7 @@ class Trainer: name=EventName.TRAIN_RAYS_PER_SEC, duration=self.world_size * self.pipeline.datamanager.get_train_rays_per_batch() - / train_t.duration, + / max(0.001, train_t.duration), step=step, avg_over_steps=True, )