pytorch
df71b3e2 - properly update _flat_weights in RNN modules (#32939)

Commit
4 years ago
properly update _flat_weights in RNN modules (#32939) Summary: Should fix https://github.com/pytorch/pytorch/issues/32346 hopefully. Now when _flat_weights list is updated, `None` elements are appended to it if some weights are missing, subsequent `setattr` calls for the missing weights should repair _flat_weights and make it suitable to use in the backend. Pull Request resolved: https://github.com/pytorch/pytorch/pull/32939 Differential Revision: D19710990 Pulled By: ngimel fbshipit-source-id: c978c7519464e94beeffa9bc33b9172854a2f298
Author
Natalia Gimelshein
Parents
Loading