lightning/pytorch_lightning/plugins
Sean Naren ee9b3fe574
[feat] pp 1/n (#5016)
* Added changes for RPC plugin

* Add missing kwargs

* Fix code format

* Loading refactors by introducing is_distributed var, fix optimizer step flow

* Add rpc guard

* Added docstrings and typing

* resolve comments

* Add additional rpc hook, refactor name of exit process hook for clarity

* remove annotation

* Modify behaviour to allow optional return, add test for rpc plugin

* resolve tests

* rename is_ddp_based

* update

* update for windows

* update

* resolve test

* code smell

* Revert back to init_ddp_connection for backwards compat

* Swap to explicit name for property

* Add missing speed parity increase for CI variability, fix call counts for child process

Co-authored-by: tchaton <thomas@grid.ai>
2020-12-08 22:02:10 +00:00
..
__init__.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
apex.py Simplify optimization Logic (#4984) 2020-12-07 12:55:49 +00:00
ddp_plugin.py [feat] pp 1/n (#5016) 2020-12-08 22:02:10 +00:00
native_amp.py Simplify optimization Logic (#4984) 2020-12-07 12:55:49 +00:00
plugin.py Allow string plugins (#4888) 2020-12-01 20:30:49 +00:00
plugin_connector.py Fix exception message for invalid custom string plugin (#4979) 2020-12-05 23:36:05 +00:00
precision_plugin.py Allow string plugins (#4888) 2020-12-01 20:30:49 +00:00
rpc_plugin.py [feat] pp 1/n (#5016) 2020-12-08 22:02:10 +00:00
sharded_native_amp_plugin.py Allow string plugins (#4888) 2020-12-01 20:30:49 +00:00
sharded_plugin.py [hotfix] ddp + manual_optimisation (#4976) 2020-12-07 19:31:54 +00:00