From 185aa1d7dc11396e946296536574fcc5dc5d9b0a Mon Sep 17 00:00:00 2001 From: Nico Kemnitz Date: Sat, 31 Jan 2026 21:14:18 +0100 Subject: [PATCH] feat(wandb_logger): print predicted value range --- deepem/train/wandb_logger.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/deepem/train/wandb_logger.py b/deepem/train/wandb_logger.py index a811dcd..d9cbec7 100644 --- a/deepem/train/wandb_logger.py +++ b/deepem/train/wandb_logger.py @@ -129,8 +129,13 @@ def log_images(self, arr = self.to_array(vec.select(0, 0), padsz=padsz) else: arr = self.to_array(torch.sigmoid(preds[key]), padsz=padsz) - logs.append(wandb.Image(arr, caption=f"{key} prediciton")) - + amin, amax = arr.aminmax() + logs.append( + wandb.Image( + arr, + caption=f"{key} prediction (low: {amin.item():.2f}, high: {amax.item():.2f})", + ) + ) # Label if key in ["affinity", "long_range", "embedding", "mitochondria_embedding"]: seg = sample[key][0,0,...].cpu().numpy().astype('uint32')