Update cli_tools.py

IBM · Jan 27, 2025 · 6bc6293 · 6bc6293
1 parent 65856ab
commit 6bc6293
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py
@@ -162,7 +162,6 @@ def write_on_batch_end(self, trainer, pl_module, prediction, batch_indices, batc
         elif isinstance(prediction, tuple):
             pred_batch, filename_batch = prediction
             for prediction, file_name in zip(torch.unbind(pred_batch, dim=0), filename_batch, strict=False):
-                print(prediction, file_name, output_dir, trainer.out_dtype)
                 save_prediction(prediction, file_name, output_dir, dtype=trainer.out_dtype)
         else:
             raise TypeError(f"Unknown type for prediction{type(prediction)}")