Rohan Varma a275491c6f [Reland] load_state_dict post hook (#77392) ... Reland of https://github.com/pytorch/pytorch/pull/76823 with fixes to call `__setstate__` for softmax/softmin/logsoftmax as per discussion with @albanD and @jbschlosser. Original description: Implements `register_load_state_dict_post_hook` API as discussed in https://github.com/pytorch/pytorch/issues/75287. Unittests cover: - Ensuring hooks are called with the correct module - Hook is called with `IncompatibleKeys` field - If hook modifies this, load_state_dict returns the modified result Pull Request resolved: https://github.com/pytorch/pytorch/pull/77392 Approved by: https://github.com/jbschlosser		2022-05-14 06:06:23 +00:00
..
api	[Reland] load_state_dict post hook (#77392)	2022-05-14 06:06:23 +00:00
jit	Typo fix in generated module name (#76880)	2022-05-09 00:51:58 +00:00
__init__.py	Convert DDP parameters to ReplicatedTensor during forward pass.	2022-04-18 03:27:23 +00:00
functional.py	Make distributed raise ImportError when not available	2022-04-20 13:05:18 +00:00