lightning/tests/utilities
thomas chaton e147127c0e
[feat] Add better support for predict + ddp 2/3 (#7215)
* wip

* update

* update

* update

* update

* update

* typo

* update on comments

* update

* update

* update

* update

* update changelog

* update

* Fix merge

* Fix merge

* move code

* resolve test

* add extra test

* add an extra test

* update on comments

* add typing

* resolve flake8

* Refactor and Docs

* Fix tests

* Fix tests

* Fix tests

* Duplicate

* Fix tests

* resolve bug

* update

* update on comments

* update

* update changelog

* update

* update

* remove tpu

* resolve flake8

* update on comments

* update on comments

* update on comment

* resolve flake8

* add a cpu test for predict

* add None test

* update

* Update CHANGELOG.md

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* resolve tests

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-04-27 08:46:45 -04: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 Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
test_apply_func.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_apply_func_torchtext.py fix importing torchtext batch (#6365) 2021-03-05 21:39:52 +01:00
test_argparse.py Fix argparse docs (#7148) 2021-04-22 02:13:00 +02:00
test_cli.py Changed basic_examples to use `LightningCLI` (#6862) 2021-04-15 15:01:16 +00:00
test_distributed.py [fix] Better support for rank_zero_only setting for SLURM and torchelastic (#6802) 2021-04-07 12:25:13 +01:00
test_dtype_device_mixin.py [feat] Add better support for predict + ddp 2/3 (#7215) 2021-04-27 08:46:45 -04: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 Typing for accelerators and plugins (#7022) 2021-04-15 16:48:16 +00:00
test_seed.py Fixed `num_sanity_val_steps` affecting reproducibility of training data shuffling (#7014) 2021-04-27 09:51:39 +00:00
test_upgrade_checkpoint.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_xla_device_utils.py [TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719) 2021-03-29 18:59:20 +01:00