tests/nn/data_parallel/test_fsdp_state_dict.py · ed7ca766436841a9372f551cc836e3620cf1aa89 · OpenDAS / fairscale

[fix] Decouple `move_params_to_cpu` from the `mixed_precision`. (#822) · ed7ca766

anj-s authored Oct 27, 2021

* remove offload dependency on fp16

* update python version for cpu tess

* run CPU tests with updated PyTorch version

* split changes

* revert tests config

* fix lint errors

* update nightly and test PyTorch versions

* skip failing multiprocess pipe test

* always skip test

* always skip test

* always skip test

* lint error

* skip unsupported versions

* improve skip message

* lint errors

* modify docs

* add tests

* fix test failures

* modify comments

* fix lint errors

* fix lint errors

ed7ca766

test_fsdp_state_dict.py 11.7 KB

Replace test_fsdp_state_dict.py