Sean Naren d01e8fdc86 [fix] Use training type plugin hook when saving (FSDP 1/n) (#6321) ... * Rely on training type plugin when saving * Add better typing to training type plugin		2021-03-04 18:09:33 +00:00
..
environments	fix duplicate console logging bug v2 (#6275)	2021-03-02 15:17:55 +05:30
precision	Simplify test for AMP plugins (#6311)	2021-03-03 08:56:57 +01:00
training_type	[fix] Use training type plugin hook when saving (FSDP 1/n) (#6321)	2021-03-04 18:09:33 +00:00
__init__.py	DeepSpeed Integration (#5954)	2021-02-17 15:23:42 -05:00
base_plugin.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00