lightning/pytorch_lightning/plugins
Sean Naren f0ab74dc2f
Expose scaler in amp plugin (#4737)
2020-11-18 22:30:47 +00:00
..
__init__.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
apex.py Sharded Accelerator 1/n: Expose clip gradients to plugins via abstract class (#4639) 2020-11-12 17:18:09 +00:00
ddp_plugin.py Move init_ddp_connection to DDP Plugin (#4407) 2020-11-18 15:49:22 -05:00
native_amp.py Expose scaler in amp plugin (#4737) 2020-11-18 22:30:47 +00:00
plugin_connector.py Enable custom apex and amp plugins (#4355) 2020-10-25 17:11:07 -04:00
precision_plugin.py Sharded Accelerator 1/n: Expose clip gradients to plugins via abstract class (#4639) 2020-11-12 17:18:09 +00:00