logging change
This commit is contained in:
parent
b927c9dc01
commit
b6096efadd
@ -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}"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user