Logs should not be hidden behind a logger.info (#6097)

06834bc3 · Lysandre Debut · GitHub · dafa296c · 06834bc3
Unverified Commit 06834bc3 authored Jul 28, 2020 by Lysandre Debut Committed by GitHub Jul 28, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -646,7 +646,7 @@ class Trainer:
        if iterator is not None:
            iterator.write(output)
        else:
-            logger.info(output)
+            print(output)

    def _prepare_inputs(
        self, inputs: Dict[str, Union[torch.Tensor, Any]], model: nn.Module