summaryrefslogtreecommitdiff
path: root/training/callbacks
diff options
context:
space:
mode:
Diffstat (limited to 'training/callbacks')
-rw-r--r--training/callbacks/wandb_callbacks.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py
index 451b0d5..6379cc0 100644
--- a/training/callbacks/wandb_callbacks.py
+++ b/training/callbacks/wandb_callbacks.py
@@ -94,7 +94,7 @@ class LogTextPredictions(Callback):
super().__init__()
def _log_predictions(
- stage: str, trainer: Trainer, pl_module: LightningModule
+ self, stage: str, trainer: Trainer, pl_module: LightningModule
) -> None:
"""Logs the predicted text contained in the images."""
if not self.ready: