logging change
This commit is contained in:
parent
b927c9dc01
commit
b6096efadd
@ -84,7 +84,7 @@ class PyTorchModelTrainer:
|
||||
# evaluation
|
||||
test_loss = self.estimate_loss(data_loaders_dictionary, self.max_n_eval_batches, "test")
|
||||
logger.info(
|
||||
f"epoch ({epoch}/{epochs}):"
|
||||
f"epoch {epoch}/{epochs}:"
|
||||
f" train loss {train_loss:.4f} ; test loss {test_loss:.4f}"
|
||||
)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user