diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 5918880d..47fd0c67 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -672,7 +672,7 @@ class Trainer(TrainerIO): def __log_vals_blacklist(self): """avoid logging some vals lightning uses to maintain state""" - blacklist = {'batch_nb', 'v_nb', 'epoch', 'gpu'} + blacklist = {'batch_nb', 'v_nb', 'gpu'} return blacklist def __run_tng_batch(self, data_batch, batch_nb):