diff --git a/zetta_utils/training/lightning/trainers/default.py b/zetta_utils/training/lightning/trainers/default.py index 52a4d841f..25995e031 100644 --- 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,