jihunchoi / recurrent-batch-normalization-pytorch

PyTorch implementation of recurrent batch normalization
243 stars 34 forks source link

Added support for multiple layers #7

Closed davda54 closed 6 years ago

davda54 commented 6 years ago

Based on a proposal by @danarte in issue #6.

In addition to this his suggestion, forwarding the hidden states has to be changed, too. Only a part of the hidden state has to be given to each layer. As I had to change the initialization of hidden states, I suggest initializing it from a uniform distribution instead of zeros (like in this article).