tests/nn/data_parallel/test_fsdp_optimizer_utils.py · 9474d75d4925eec3353a06d2682fce805fc8970c · OpenDAS / fairscale · GitLab

Find file Blame History Permalink

[FSDP][feature] optimizer state dict save and load (#537) · 9474d75d
Sam Shleifer authored Mar 25, 2021
```
Co-authored-by: Min Xu <24926999+min-xu-ai@users.noreply.github.com>
```
9474d75d

test_fsdp_optimizer_utils.py 4.57 KB