OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
daff263062
pytorch/torch/nn
History
zeshengzong 1b569e5490 Fix load_state_dict description (#154599) 
Fixes #141364

Fix missing description in `assign` param

## Test Result

### Before
![image](https://github.com/user-attachments/assets/5928c691-4e31-463b-aa0a-86eb8bb452e5)

### After
![image](https://github.com/user-attachments/assets/036631a2-0f20-4a71-95c3-2c0fd732293e)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154599
Approved by: https://github.com/colesbury, https://github.com/mikaylagawarecki
2025-05-30 18:08:59 +00:00
..
attention [FlexAttention] Enforce Q,K,V memory layouts for fp8 flex attention to avoid perf degradation (#153357) 2025-05-16 04:56:50 +00:00
backends
intrinsic
modules Fix load_state_dict description (#154599) 2025-05-30 18:08:59 +00:00
parallel [ddp] propagate use_python_reducer to C++ reducer (#152735) 2025-05-16 01:38:03 +00:00
qat
quantizable
quantized
utils remove allow-untyped-defs from torch/nn/utils/_expanded_weights/conv_expanded_weights.py (#154623) 2025-05-30 07:32:57 +00:00
__init__.py
_reduction.py
common_types.py [BE][Ez]: Denote common types as TypeAlias (#154527) 2025-05-29 02:00:13 +00:00
cpp.py
functional.py Add pad limit of avg_poolnd and AvgPoolnd (#152680) 2025-05-04 17:25:22 +00:00
functional.pyi.in [BE][CI][Easy] Run lintrunner on generated .pyi stub files (#150732) 2025-05-27 14:58:02 +00:00
grad.py
init.py
parameter.py
parameter.pyi remove allow-untyped-defs from torch/nn/parameter.pyi (#144654) 2025-01-13 19:02:31 +00:00