From 5e6706468602f4560c6cc317518294bfeea28a05 Mon Sep 17 00:00:00 2001 From: Olivier Date: Thu, 26 Oct 2023 22:14:01 +0200 Subject: [PATCH] :shirt: Remove print --- torch_uncertainty/routines/classification.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/torch_uncertainty/routines/classification.py b/torch_uncertainty/routines/classification.py index 9c99c71a..0277d1f5 100644 --- a/torch_uncertainty/routines/classification.py +++ b/torch_uncertainty/routines/classification.py @@ -597,8 +597,6 @@ def test_epoch_end( id_logits = torch.cat(outputs[0], 0).float().cpu() ood_logits = torch.cat(outputs[1], 0).float().cpu() - print(id_logits.shape) - id_probs = F.softmax(id_logits, dim=-1) ood_probs = F.softmax(ood_logits, dim=-1)