Adrian Wälchli 4b7c0fae00 Fix amp autocast (#6080) ... * precision fixes * add amp test model * fix test * revert * move assert to training step * fix test * fix test * remove unrelated changes * add changelog * remove unused import		2021-02-19 17:00:27 +00:00
..
__init__.py	DeepSpeed Integration (#5954)	2021-02-17 15:23:42 -05:00
apex_amp.py	[accelerator][FeatBugFix] Improve manual optimization API (#5771)	2021-02-16 16:00:35 -05:00
deepspeed_precision.py	DeepSpeed Integration (#5954)	2021-02-17 15:23:42 -05:00
mixed.py	Accelerator Refactor: Precision Plugins (#5718)	2021-01-31 13:12:02 -05:00
native_amp.py	Fix amp autocast (#6080)	2021-02-19 17:00:27 +00:00
precision_plugin.py	fix flake8 for new plugins (#5951)	2021-02-18 18:28:23 +00:00
sharded_native_amp.py	Hardware specific parts of Accelerator Refactoring (#5719)	2021-02-01 08:34:59 -05:00
tpu_bfloat.py	fix flake8 for new plugins (#5951)	2021-02-18 18:28:23 +00:00