tests/nn/data_parallel/test_fsdp_grad_scaler.py · 5ecac15adeed3cc3452167f507bc9fa773608c35 · OpenDAS / fairscale

[test] FSDP: add the failing test for #421 (#453) · 5ecac15a

Min Xu authored Mar 01, 2021



* [test] FSDP: add the failing test for #421

* skip on 1.5

* better skipping

* Update tests/nn/data_parallel/test_fsdp_grad_scaler.py
Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
Co-authored-by: Sam Shleifer <sshleifer@gmail.com>

5ecac15a

test_fsdp_grad_scaler.py 1.29 KB

Replace test_fsdp_grad_scaler.py