From ba0a32c2ae86724a09b74091633dbb452a5705e7 Mon Sep 17 00:00:00 2001 From: Ryan Wong Date: Tue, 12 Nov 2019 03:58:32 +0000 Subject: [PATCH] fixed issue where callback_metrics was replaced instead of updated (#492) --- pytorch_lightning/trainer/evaluation_loop_mixin.py | 2 +- pytorch_lightning/trainer/train_loop_mixin.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/evaluation_loop_mixin.py b/pytorch_lightning/trainer/evaluation_loop_mixin.py index c2ba5d16..8583805a 100644 --- a/pytorch_lightning/trainer/evaluation_loop_mixin.py +++ b/pytorch_lightning/trainer/evaluation_loop_mixin.py @@ -138,7 +138,7 @@ class TrainerEvaluationLoopMixin(object): self.log_metrics(log_metrics, {}) # track metrics for callbacks - self.callback_metrics = callback_metrics + self.callback_metrics.update(callback_metrics) # hook model.on_post_performance_check() diff --git a/pytorch_lightning/trainer/train_loop_mixin.py b/pytorch_lightning/trainer/train_loop_mixin.py index 306416db..68c34cf4 100644 --- a/pytorch_lightning/trainer/train_loop_mixin.py +++ b/pytorch_lightning/trainer/train_loop_mixin.py @@ -258,7 +258,7 @@ class TrainerTrainLoopMixin(object): all_log_metrics = {k: v for d in all_log_metrics for k, v in d.items()} # track all metrics for callbacks - self.callback_metrics = {k: v for d in all_callback_metrics for k, v in d.items()} + self.callback_metrics.update({k: v for d in all_callback_metrics for k, v in d.items()}) return 0, grad_norm_dic, all_log_metrics