From 4c7f720fe42278ec27e87bb1c3d5fcc6f1f822c2 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Thu, 9 Jun 2022 22:35:04 +0200
Subject: Use predict in log text callback
---
training/callbacks/wandb_callbacks.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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 = [
--
cgit v1.2.3-70-g09d2