diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-09 22:35:04 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-09 22:35:04 +0200 |
commit | 4c7f720fe42278ec27e87bb1c3d5fcc6f1f822c2 (patch) | |
tree | 5618a99f310981ab01c464bc8605c09abe26bba9 | |
parent | 9353a39a18d0542afc177cd134f33f0756820a7d (diff) |
Use predict in log text callback
-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 = [ |