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

updates

This commit is contained in:
mehrad 2019-05-28 18:12:31 -07:00
parent bb90a35bc0
commit eb10a788b0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
decanlp/models/common.py
View File

@ -412,7 +412,7 @@ class Embedding(nn.Module):
def forward(self, x, lengths=None, device=-1):
if self.pretrained_embeddings is not None:
pretrained_embeddings = self.pretrained_embeddings[0](x).to(x.device).detach()
pretrained_embeddings = self.pretrained_embeddings[0](x.cpu()).to(x.device).detach()
else:
pretrained_embeddings = None
if self.trained_embeddings is not None: