From 4c7f720fe42278ec27e87bb1c3d5fcc6f1f822c2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm 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(-) (limited to 'training') 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