lightning/pytorch_lightning/plugins
ananthsub 6878f3bf4e
Enable DDP Plugin to pass through args to LightningDistributedDataParallel (#4382)
* Update ddp_plugin.py

* Update ddp_plugin.py

* Update ddp_plugin.py

* Update test_ddp_plugin.py

* Update pytorch_lightning/plugins/ddp_plugin.py

* Update pytorch_lightning/plugins/ddp_plugin.py

* Fixed imports, make ddp_kwargs protected

Co-authored-by: SeanNaren <sean.narenthiran@gmail.com>
2020-10-27 12:27:59 +00:00
..
__init__.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
apex.py Enable custom apex and amp plugins (#4355) 2020-10-25 17:11:07 -04:00
ddp_plugin.py Enable DDP Plugin to pass through args to LightningDistributedDataParallel (#4382) 2020-10-27 12:27:59 +00:00
native_amp.py Enable custom apex and amp plugins (#4355) 2020-10-25 17:11:07 -04:00
plugin_connector.py Enable custom apex and amp plugins (#4355) 2020-10-25 17:11:07 -04:00