projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8f11147) | patch
use all_weights instead of _parameters in _flat_weights in rnn (#15766)
authorNatalia Gimelshein <ngimelshein@nvidia.com>
Tue, 8 Jan 2019 17:45:52 +0000 (09:45 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 8 Jan 2019 17:48:36 +0000 (09:48 -0800)
commit461dc9a28b132f032e1ae81bfdab27e9d3770fe9
treec5eb6b0280f5c800dc1ba0c960ea5f0e2bc31640tree | snapshot
parent8f11147d43386acec2642a23cf08710ee0ab1af5commit | diff
use all_weights instead of _parameters in _flat_weights in rnn (#15766)

Summary:
Fixes #15749
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15766

Differential Revision: D13592320

Pulled By: soumith

fbshipit-source-id: 6c3805f576c3df5a2da8bef1e4305eda379718df
test/test_nn.py diff | blob | history
torch/nn/modules/rnn.py diff | blob | history
Domain: Machine Learning / ML Framework;