wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed callback metrics ddp bug

Browse Source
This commit is contained in:
William Falcon
2019-10-09 12:53:33 -04:00
parent 5f1f3f6acc
commit 6e0a562ecb
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+3
View File
@@ -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
# ---------------