pytorch
ce101fed - [PyPer] copy-free freeze_module (#65118)

Commit
4 years ago
[PyPer] copy-free freeze_module (#65118) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65118 Cloning the module can increase memory use. By freezing the module directly without cloning it first, we can avoid this memory usage increase. Reviewed By: eellison, movefast1990 Differential Revision: D30955053 fbshipit-source-id: 2feb738eddcf66aa68c92bf695cc05b57bd990f0
Author
Hao Lu
Parents
Loading