Skip to content

GitLab

Unverified Commit 06834bc3 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

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

parent dafa296c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ 06834bc3
...@@ -646,7 +646,7 @@ class Trainer: ...@@ -646,7 +646,7 @@ class Trainer:
if iterator is not None: if iterator is not None:
iterator.write(output) iterator.write(output)
else: else:
logger.info(output) print(output)
def _prepare_inputs( def _prepare_inputs(
self, inputs: Dict[str, Union[torch.Tensor, Any]], model: nn.Module self, inputs: Dict[str, Union[torch.Tensor, Any]], model: nn.Module
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment