Rooba
/
genienlp
mirror of https://github.com/stanford-oval/genienlp.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Multilayer LSTM bug fixed

This commit is contained in:
Sina 2020-03-02 16:57:49 -08:00
parent ea078d8e46
commit 11f37c590b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
genienlp/models/identity_encoder.py
View File

@ -54,7 +54,7 @@ class IdentityEncoder(nn.Module):
if self.args.rnn_layers > 0 and self.args.rnn_zero_state == 'average':
self.pool = LinearFeedforward(args.dimension, args.dimension, 2 * args.rnn_dimension * args.rnn_layers,
dropout=args.dropout_ratio)
self.norm = LayerNorm(2 * args.rnn_dimension)
self.norm = LayerNorm(2 * args.rnn_dimension * args.rnn_layers)
else:
self.pool = None
self.norm = None