From 6e0a562ecb094ef28c5b897a68af071e7d8d2da3 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 9 Oct 2019 12:53:33 -0400
Subject: [PATCH] fixed callback metrics ddp bug

---
 pytorch_lightning/trainer/trainer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
index 74c69074..7b3b8029 100644
--- a/pytorch_lightning/trainer/trainer.py
+++ b/pytorch_lightning/trainer/trainer.py
@@ -1285,6 +1285,9 @@ class Trainer(TrainerIO):
             nb_gpus = self.num_gpus
             callback_metrics = reduce_distributed_output(callback_metrics, nb_gpus)
 
+        for k, v in callback_metrics.items():
+            callback_metrics[k] = v.item()
+
         # ---------------
         # EXTRACT PROGRESS BAR KEYS
         # ---------------