chaton 102fa9ee7d [BUGFIX] AMP + Precision unscale grad (#4441) ... * move unscale within Native plugin * remove gradient tracking from lightning backward * forgot trainer.fit * typo * update * cleanup * set to 1.6 * typo * skip if below 1.6 strict * update changelog * remove useless code * Update tests/plugins/test_amp_plugin.py Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> * Update tests/plugins/test_amp_plugin.py Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> * update changelog * Update CHANGELOG.md Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> Co-authored-by: Jeff Yang <ydcjeff@outlook.com>		2020-11-02 16:36:48 +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	[BUGFIX] AMP + Precision unscale grad (#4441)	2020-11-02 16:36:48 +00:00
plugin_connector.py	Enable custom apex and amp plugins (#4355)	2020-10-25 17:11:07 -04:00