shuyingsunshine21 8538c1f61e Accelerator model state dict (#7474) ... * Fix some test errors Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: * checkpoint consolidation * Update ddp_spawn.py * Update test_metric_result_integration.py * Update test_results.py * Update utils.py * Update utils.py * Update test_all_gather_grad.py * Update test_all_gather_grad.py * Update test_results.py * Revert "Update test_results.py" This reverts commit 9d4a2b891d. * Revert "Merge pull request #1 from shuyingsunshine21/shuyingsunshine21-checkpoint_consolidate" This reverts commit c5053da789, reversing changes made to 0d23d75bc9. * Revert "Update test_all_gather_grad.py" This reverts commit 0d23d75bc9. * Revert "Update utils.py" This reverts commit 70fe5da9c6. * Revert "Update utils.py" This reverts commit a9aae99f6e. * Revert "Update test_results.py" This reverts commit ea74906878. * Revert "Update test_metric_result_integration.py" This reverts commit bf70e431b3. * Revert "Update ddp_spawn.py" This reverts commit f17210183b. * Revert "checkpoint consolidation" This reverts commit 536c1323b0. * Revert "Revert "checkpoint consolidation"" This reverts commit 3a9fde915a. * Revert "Revert "Revert "checkpoint consolidation""" This reverts commit 7a369f47e1. * Revert "Revert "Update ddp_spawn.py"" This reverts commit 8222dc98ea. * Revert "Revert "Update test_metric_result_integration.py"" This reverts commit 6c095b2370. * Revert "Revert "Update test_results.py"" This reverts commit 250d0aaaa2. * Revert "Revert "Update utils.py"" This reverts commit 8651d54d79. * Revert "Revert "Update test_all_gather_grad.py"" This reverts commit dcdcd29731. * modify distributed environment to make test pass * modify model state dict to training type plugin * remove changes * add changelog * fixing isort for pre-commit failure * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Address code review Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: SeanNaren <sean@grid.ai>		2021-05-11 16:39:04 +01:00
..
__init__.py	PoC: Accelerator refactor (#5743)	2021-02-12 15:48:56 -05:00
accelerator.py	Accelerator model state dict (#7474)	2021-05-11 16:39:04 +01:00
cpu.py	release 1.3.0 (#7404)	2021-05-06 15:05:35 -04:00
gpu.py	fix display bug (#7395)	2021-05-10 11:26:15 +08:00
tpu.py	Use `torch.nn.utils.clip_grad_norm_` and add `clip_grad_by_value` support for TPU (#7025)	2021-05-07 16:41:39 +00:00