Lucas Pasqualin b7fac76fc2 [DCP fixes for _load_state_dict_keys and supports nested keys (#123679) ... Fixes some issues with `_load_state_dict_keys`, including: * updates broken test, which was failing due to incorrect parameters * adds support for specifying nested keys e.g. (load state dict keys can now specify `something like "optimizer.state"`, which loads all keys under `optimzier.state`. * updates call site to use the private implementation of `_load_state_dict`, which properly handles empty state dicts (otherwise the keys are ignored) Big shout out to @diego-urgell who not only identified current issues, but recommended the right solutions! Pull Request resolved: https://github.com/pytorch/pytorch/pull/123679 Approved by: https://github.com/diego-urgell, https://github.com/wz337		2024-04-11 20:52:06 +00:00
..
_internal	[DCP fixes for _load_state_dict_keys and supports nested keys (#123679)	2024-04-11 20:52:06 +00:00
__init__.py
_comparison.py	[BE]: Apply RUF025 dict.fromkeys preview rule (#118637)	2024-01-30 20:46:54 +00:00
_creation.py	additional support for float8_e4m3fnuz and _e5m2fnuz (#115214)	2024-01-22 18:33:41 +00:00