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

Fix embedding layer implementation 

Embedding dropout parameter is not used
This commit is contained in:
Victoria X Lin 2018-09-11 23:06:58 -07:00 committed by Bryan McCann
parent 4523bcabe8
commit c0e4f7083a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/common.py
View File

@ -333,7 +333,7 @@ class Embedding(nn.Module):
if self.project: if self.project:
self.projection = Feedforward(dimension, trained_dimension) self.projection = Feedforward(dimension, trained_dimension)
dimension = trained_dimension dimension = trained_dimension
self.dropout = nn.Dropout(0.2) self.dropout = nn.Dropout(dropout)
self.dimension = dimension self.dimension = dimension
def forward(self, x, lengths=None, device=-1): def forward(self, x, lengths=None, device=-1):