lightning/tests/tests_pytorch/plugins
Rohit Gupta 0f6caffa57
Fix deepspeed default precision plugin `amp_level` to O2 (#13897)
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
2022-07-29 20:36:51 +00:00
..
environments Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
precision Fix deepspeed default precision plugin `amp_level` to O2 (#13897) 2022-07-29 20:36:51 +00: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