👕 Remove print

ENSTA-U2IS-AI · Oct 26, 2023 · 5e67064 · 5e67064
1 parent 2b96285
commit 5e67064
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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)