lightning/pytorch_lightning/plugins
Sean Naren efda48faab
Disable CPU Offload as default for DeepSpeed (#6262)
* Change default for CPU offload to false for best throughput/memory efficiency

* Add changelog

* default

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-03-02 07:53:12 +00:00
..
environments remove legacy plugins (#5950) 2021-02-16 19:20:58 +00:00
precision simplify skip-if tests >> 0/n (#5920) 2021-03-01 12:17:09 +00:00
training_type Disable CPU Offload as default for DeepSpeed (#6262) 2021-03-02 07:53:12 +00:00
__init__.py DeepSpeed Integration (#5954) 2021-02-17 15:23:42 -05:00
base_plugin.py Add mypy typing to precision plugins. (#6149) 2021-02-26 14:27:16 +01:00