lightning/tests/utilities
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
..
__init__.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
distributed.py Fixed PYTHONPATH for ddp test model (#4528) 2020-12-05 20:09:47 +00:00
test_all_gather_grad.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_apply_func.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_apply_func_torchtext.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_argparse_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
test_dtype_device_mixin.py Prune EvalModelTemplate from callbacks and utilities (#6018) 2021-02-16 19:59:57 +00:00
test_imports.py fix import and typo in AMP (#4871) 2020-11-26 23:45:52 +01:00
test_memory.py Make move_metrics_to_cpu work recursively (#6007) 2021-02-16 21:52:42 +00:00
test_parsing.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_seed.py fix imports / isort / flake8 2021-01-26 14:57:34 +01:00
test_upgrade_checkpoint.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_xla_device_utils.py fix deprecated call (#6005) 2021-02-16 12:11:13 +01:00