fix: addendum to #534

ZettaAI · Nov 8, 2023 · d3b30ba · d3b30ba
1 parent d78b69e
commit d3b30ba
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/zetta_utils/training/lightning/trainers/default.py b/zetta_utils/training/lightning/trainers/default.py
@@ -227,7 +227,6 @@ def unwrap_forward(pl_module: pl.LightningModule, name: str) -> None:
         delattr(model, "__forward__")
         del wrapped_forward
 
-    @pl.utilities.rank_zero.rank_zero_only
     def on_validation_batch_start(
         self,
         trainer: pl.Trainer,