tests/nn/misc/test_flatten_params_wrapper.py · 03f3e833ad58cf0ace5bdf6be112667e99cf9cf9 · OpenDAS / fairscale

[fix] a bug in loading flatten state dict (#1025) · 03f3e833

Min Xu authored Jul 12, 2022



- Thanks to Alexei for spotting this.
- Added triggering test case.
- enhanced tests with more comments etc.
Co-authored-by: Min Xu <min.xu.public@gmail.com>

03f3e833

test_flatten_params_wrapper.py 16 KB

Replace test_flatten_params_wrapper.py