👌 Remove print() in RegressionRoutine

ENSTA-U2IS-AI · Sep 4, 2024 · 2dee70b · 2dee70b
1 parent 4e3a055
commit 2dee70b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torch_uncertainty/routines/regression.py b/torch_uncertainty/routines/regression.py
@@ -155,7 +155,6 @@ def training_step(
         self, batch: tuple[Tensor, Tensor], batch_idx: int
     ) -> STEP_OUTPUT:
         inputs, targets = self.format_batch_fn(batch)
-        print(inputs.shape, targets.shape)
 
         if self.one_dim_regression:
             targets = targets.unsqueeze(-1)