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

Update logging.py (#602)

Browse Source
This commit is contained in:
ctlaltdefeat
2019-12-07 10:12:33 -05:00
committed by William Falcon
parent 2f01c03b38
commit 58cc6e13b9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/logging.py
+2 -1
View File
@@ -87,7 +87,8 @@ class TrainerLoggingMixin(ABC):
callback_metrics = self.reduce_distributed_output(callback_metrics, num_gpus)
for k, v in callback_metrics.items():
callback_metrics[k] = v.item()
if isinstance(v, torch.Tensor):
callback_metrics[k] = v.item()
# ---------------
# EXTRACT PROGRESS BAR KEYS