Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove fn check for ipu output (#7915)

This commit is contained in:
Sean Naren 2021-06-10 12:35:32 +01:00 committed by GitHub
parent 580a3b5e32
commit 07b69231ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/connectors/logger_connector/result_new.py
View File

@ -264,8 +264,6 @@ class ResultCollection(dict):
if loss is not None: if loss is not None:
if not isinstance(loss, torch.Tensor): if not isinstance(loss, torch.Tensor):
raise ValueError(f"`Result.minimize` must be a `torch.Tensor`, found: {loss}") raise ValueError(f"`Result.minimize` must be a `torch.Tensor`, found: {loss}")
if loss.grad_fn is None:
raise RuntimeError("`Result.minimize` must have a `grad_fn`")
self._minimize = loss self._minimize = loss
@property @property