Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/plugins
History
Sean Naren 432e5637d6
Expose DeepSpeed FP16 parameters due to loss instability (#6115) 
* Expose deepspeed config parameters to init function due to instability in parameters

* See if tests can run on normal CI, without special tests

* Add changelog

* Update pytorch_lightning/plugins/training_type/deepspeed.py

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

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
 		2021-02-21 21:43:11 +01:00
..
__init__.py enable ddp as a plugin (#4285) 2020-10-22 05:15:51 -04:00
test_amp_plugin.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_apex_plugin.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_deepspeed_plugin.py Expose DeepSpeed FP16 parameters due to loss instability (#6115) 2021-02-21 21:43:11 +01:00
test_rpc_plugin.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_rpc_sequential_plugin.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
test_sharded_plugin.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00