diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-02 01:01:51 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-02 01:01:51 +0200 |
| commit | e5eca28438cd17d436359f2c6eee0bb9e55d2a8b (patch) | |
| tree | c8d77aa0a77d31f63dc2b81eee6d9c5f37e0e2f1 /training/callbacks | |
| parent | 04876c35a1eeea68794745f1ca113754c0d97757 (diff) | |
Fixed wandb callback bug, add word_pieces arg to iam paragraph
Diffstat (limited to 'training/callbacks')
| -rw-r--r-- | training/callbacks/wandb_callbacks.py | 2 |
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: |