Remove print output

rapidsai · Oct 25, 2024 · 5cfa3ce · 5cfa3ce
1 parent 03f717c
commit 5cfa3ce
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/crossfit/backend/torch/model.py b/crossfit/backend/torch/model.py
@@ -118,7 +118,6 @@ def _add_numeric_column(
     )
     del all_outputs_ls
     del loader
-    print("outputs", outputs, flush=True)
     cleanup_torch_cache()
     if len(outputs.shape) == 1:
         df[pred_output_col] = cudf.Series(outputs, index=_index)