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

This commit is contained in:
Lysandre Debut
2020-07-28 12:44:25 -04:00
committed by GitHub
parent dafa296c95
commit 06834bc332
+1 -1
View File
@@ -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