Justus Schock 634d83134f Add AMP for validation, prediction and testing (#6565) ... * Add Tests for val and test-steps * Add native AMP * pep8 tests * pep8 plugin * changelog		2021-03-20 23:15:49 +00:00
..
__init__.py	DeepSpeed Integration (#5954)	2021-02-17 15:23:42 -05:00
apex_amp.py	[bug] Update broadcast + reduce decision ModelCheckpoint] (#6410)	2021-03-14 17:14:27 +00:00
deepspeed_precision.py	simplify skip-if tests >> 0/n (#5920)	2021-03-01 12:17:09 +00:00
mixed.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00
native_amp.py	Add AMP for validation, prediction and testing (#6565)	2021-03-20 23:15:49 +00:00
precision_plugin.py	Update Gradient Clipping for TPU Accelerator (#6576)	2021-03-20 01:02:57 +05:30
sharded_native_amp.py	[Fix] Call clip gradients if clip val greater than 0 (#6330)	2021-03-04 19:45:58 +00:00
tpu_bfloat.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00