logging change

This commit is contained in:
Yinon Polak 2023-03-13 00:35:14 +02:00
parent b927c9dc01
commit b6096efadd

View File

@ -84,7 +84,7 @@ class PyTorchModelTrainer:
# evaluation # evaluation
test_loss = self.estimate_loss(data_loaders_dictionary, self.max_n_eval_batches, "test") test_loss = self.estimate_loss(data_loaders_dictionary, self.max_n_eval_batches, "test")
logger.info( logger.info(
f"epoch ({epoch}/{epochs}):" f"epoch {epoch}/{epochs}:"
f" train loss {train_loss:.4f} ; test loss {test_loss:.4f}" f" train loss {train_loss:.4f} ; test loss {test_loss:.4f}"
) )