diff options
| -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 c9d50d5..f98cfa5 100644 --- a/training/callbacks/wandb_callbacks.py +++ b/training/callbacks/wandb_callbacks.py @@ -107,7 +107,7 @@ class LogTextPredictions(Callback): imgs, labels = samples imgs = imgs.to(device=pl_module.device) - logits = pl_module(imgs) + logits = pl_module.predict(imgs) mapping = pl_module.mapping data = [ |