Prevent loss to be moved to the cpu before backward call.#9308
Merged
tchaton merged 7 commits intomasterfrom bugfix/9296_move_metrics_to_cpuSep 3, 2021
+25-3
Commits
Commits on Sep 3, 2021
- committed
- committed
- committed
Merge branch 'bugfix/9296_move_metrics_to_cpu' of https://github.com/PyTorchLightning/pytorch-lightning into bugfix/9296_move_metrics_to_cpu
committed- committed
- committed
- committed
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}