Yanli Zhao 956bd03808 add ignored_states to FSDP/fully_shard (#102056) ... Add 'ignored_states' that accepts either a list of ignored_parameters or a list of nn modules for FSDP model wrapper and fully_shard composable APIs, it is recommended to use 'ignored_states' over 'ignored_modules' moving forward Pull Request resolved: https://github.com/pytorch/pytorch/pull/102056 Approved by: https://github.com/awgu		2023-05-24 18:36:48 +00:00
..
__init__.py	Add global registry to composable API contract (#90579)	2022-12-10 22:41:10 +00:00
checkpoint_activation.py	[BE]: Enable ruff rule TRY302 and apply fixes (#101874)	2023-05-19 17:30:52 +00:00
contract.py	[Composable] add checking key for check_fqn function (#98961)	2023-04-13 03:16:14 +00:00
fully_shard.py	add ignored_states to FSDP/fully_shard (#102056)	2023-05-24 18:36:48 +00:00
replicate.py	[replicate] support simpler device_id (#100217)	2023-05-04 21:06:04 +00:00