lightning/tests/tests_pytorch/plugins
Carlos Mocholí e0c2c3e677
Clean up fairscale imports (#14476)
2022-09-01 18:08:40 +02:00
..
environments Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
precision Clean up fairscale imports (#14476) 2022-09-01 18:08:40 +02:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_amp_plugins.py Fix deepspeed default precision plugin `amp_level` to O2 (#13897) 2022-07-29 20:36:51 +00:00
test_checkpoint_io_plugin.py Add support for async checkpointing (#13658) 2022-07-26 21:13:19 +05:30
test_cluster_integration.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_double_plugin.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00