pytorch
817eb94f - Speed up module constructor by avoiding module.__setattr__ (#77098)

Commit
3 years ago
Speed up module constructor by avoiding module.__setattr__ (#77098) Module's overridden `__setattr__` has special handling for parameters, submodules, and buffers, resulting in reduces performance for its default attributes (which are not of those types). Setting them directly results in a significant improvement for module construction speed (~10x). Pull Request resolved: https://github.com/pytorch/pytorch/pull/77098 Approved by: https://github.com/jbschlosser
Author
Committer
Parents
Loading