Fix bug

carmocca · carmocca · commit 5c8ce17cceea · 2021-06-17T02:52:29.000+02:00
diff --git a/pytorch_lightning/trainer/connectors/logger_connector/result.py b/pytorch_lightning/trainer/connectors/logger_connector/result.py
@@ -83,6 +83,7 @@ def reduce_fx(self, reduce_fx: Union[str, Callable]) -> None:
             ' Please, open an issue in `https://github.com/PyTorchLightning/pytorch-lightning/issues`.'
             f' Found: {reduce_fx}'
         )
+        self._reduce_fx = reduce_fx
         if isinstance(reduce_fx, str):
             reduce_fx = reduce_fx.lower()
             if reduce_fx == 'avg':

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,7 @@ def reduce_fx(self, reduce_fx: Union[str, Callable]) -> None:`
`83`	`83`	' Please, open an issue in `https://github.com/PyTorchLightning/pytorch-lightning/issues`.'
`84`	`84`	`f' Found: {reduce_fx}'`
`85`	`85`	`)`
	`86`	`+ self._reduce_fx = reduce_fx`
`86`	`87`	`if isinstance(reduce_fx, str):`
`87`	`88`	`reduce_fx = reduce_fx.lower()`
`88`	`89`	`if reduce_fx == 'avg':`