Commit Graph

2 Commits

Author SHA1 Message Date
Rio H ba8053492f
Deprecate LightningModule.model_size (#8495)
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-07-30 13:53:40 +00:00
Dusan Drevicky c9fde04947
Make move_metrics_to_cpu work recursively (#6007)
* Propagate to_cpu flag down the recursion chain

* Refactor

* Add test

* Update CHANGELOG

* Update tests/utilities/test_memory.py

Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-02-16 21:52:42 +00:00