lightning

Author	SHA1	Message	Date
Kaushik B	ef7d41692c	Add `ddp_*_find_unused_parameters_false` to Plugins Registry. (#8483 ) Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>	2021-07-24 04:02:54 +00:00
Kaushik B	78a14a3f56	Add `tpu_spawn_debug` to plugin registry (#7933 )	2021-06-15 22:32:51 +00:00
Sean Naren	f7459f5328	DeepSpeed Infinity Update (#7234 ) * Update configs to match latest API * Ensure we move the entire model to device before configure optimizer is called * Add missing param * Expose parameters * Update references, drop local rank as it's now infered from the environment variable * Fix ref * Force install deepspeed 0.3.16 * Add guard for init * Update pytorch_lightning/plugins/training_type/deepspeed.py Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * Revert type checking * Install master for CI for testing purposes * Update CI * Fix tests * Add check * Update versions * Set precision * Fix * See if i can force upgrade * Attempt to fix * Drop * Add changelog Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>	2021-06-14 16:38:28 +00:00
Kaushik B	e21b7a62d7	Add ddp_find_unused_parameters_false to Registry (#7224 )	2021-05-04 22:40:00 +00:00
Kaushik B	6d7c6d6403	Update Accelerator Connector for Registry (#7214 )	2021-05-03 21:03:21 +00:00
Kaushik B	832a03af7c	Add Training Type Plugins Registry (#6982 ) Co-authored-by: Sean Naren <sean@grid.ai> Co-authored-by: thomas chaton <thomas@grid.ai>	2021-04-16 18:01:56 +05:30