I have recently come across this note, which proposes better initialization scheme for sub-pixel/pixel-shuffle
https://arxiv.org/abs/1707.02937 92
Here is a sample implementation I found in Lasagne (https://github.com/Lasagne/Lasagne/issues/862 83)
How can we implement the same in pytorch ?
I’m new to pytorch. I have gone through forums and found that we can custom initialize weights with,
for m in self.modules():
    if isinstance(m, nn.Conv2d):
        m.weight.data.copy_(some_weights)

It would be great if someone can provide a sample implementation the authors have suggested in the paper.
Thanks