pytorch
00c6b903 - Fix in documentation of convolutional modules (#30079)

Commit
5 years ago
Fix in documentation of convolutional modules (#30079) Summary: I noticed the description of the initialization of convolutional modules is inconsistent with the actual implementation. There are two such cases: 1) `k` in the initialization of ConvTranspose modules is not dependent on the input channels but on the output channels (`kaiming_uniform_` uses the size of the second dimension of `weight` which is transposed in the first two dimensions). 2) Both the normal convolutions and the transposed ones use `k` divided by `groups`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/30079 Differential Revision: D19698511 Pulled By: ezyang fbshipit-source-id: 1ba938fbbd97663eaf29fd1245872179d2761fff
Author
Parents
Loading