mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Fixes https://github.com/pytorch/pytorch/issues/148242 ignored_params won't be moved to devices in full_shard(), update docstring. Add unit test `test_move_states_to_device_ignored_param_device` to show that ignored_params won't be moved during full_shard(), but would be after `model.cuda()` Pull Request resolved: https://github.com/pytorch/pytorch/pull/149074 Approved by: https://github.com/awgu |
||
|---|---|---|
| .. | ||
| fsdp | ||
| test_composability | ||
| test_checkpoint.py | ||
| test_contract.py | ||
| test_replicate_with_compiler.py | ||
| test_replicate.py | ||