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)