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
Ethan Harris 03bb389b21
Fix double precision + ddp_spawn (#6924) 
* Initial fix

* Initial fix

* Initial fix

* Updates

* Updates

* Update typing and docs

* Undo accidental refactor

* Remove unused imports

* Add DDP double precision test

* Remove unused variable

* Update CHANGELOG.md

* Fix test

* Update tests

* Formatting

* Revert bad change

* Add back changes

* Correct wrapping order

* Improve unwrapping

* Correct wrapping order

* Fix... finally

* Respond to comments

* Drop ddp test

* Simplify ddp spawn test

* Simplify ddp spawn test

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
 		2021-06-01 15:21:17 +00:00
..
environments Add kubeflow cluster environment (#7300) 2021-05-17 09:05:24 +01:00
__init__.py enable ddp as a plugin (#4285) 2020-10-22 05:15:51 -04:00
test_amp_plugins.py [bugfix] Apex never instantiated. (#7274) 2021-04-30 13:16:28 -04:00
test_cluster_integration.py Set `num_nodes` and `sync_batchnorm` From Trainer for Manually Passed Training Type Plugin (#7026) 2021-05-08 11:25:51 +00:00
test_custom_plugin.py Add typings for evaluation_loop.py and remove some dead code (#7015) 2021-04-15 07:36:04 +00:00
test_ddp_fully_sharded_with_full_state_dict.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
test_ddp_plugin.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00
test_ddp_plugin_with_comm_hook.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_ddp_spawn_plugin.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00
test_deepspeed_plugin.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00
test_double_plugin.py Fix double precision + ddp_spawn (#6924) 2021-06-01 15:21:17 +00:00
test_plugins_registry.py Add ddp_find_unused_parameters_false to Registry (#7224) 2021-05-04 22:40:00 +00:00
test_rpc_plugin.py Clean up environment access in plugins (#6941) 2021-04-13 20:07:40 +02:00
test_rpc_sequential_plugin.py Remove legacy support for the magic `log`/`progress_bar` keys in dict returns (#6734) 2021-03-31 00:28:04 +02:00
test_sharded_plugin.py Fix ShardedDataParallel has no attribute require_backward_grad_sync (#6915) 2021-04-10 16:14:37 +00:00
test_single_device_plugin.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00
test_tpu_spawn.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00