From eb10a788b09589fc5fbb63b9165ace9e264dac40 Mon Sep 17 00:00:00 2001 From: mehrad Date: Tue, 28 May 2019 18:12:31 -0700 Subject: [PATCH] updates --- decanlp/models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/decanlp/models/common.py b/decanlp/models/common.py index e7ae6633..c4a4dda4 100644 --- a/decanlp/models/common.py +++ b/decanlp/models/common.py @@ -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: