From cb3cb8bc5cf5c2d0f083a6e5f1608f091682f405 Mon Sep 17 00:00:00 2001 From: yh Date: Thu, 25 Jul 2019 16:51:59 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DBertEmbedding=E7=9A=84weight?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/embeddings/embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/embeddings/embedding.py b/fastNLP/embeddings/embedding.py index a9f228fb..9447c6ad 100644 --- a/fastNLP/embeddings/embedding.py +++ b/fastNLP/embeddings/embedding.py @@ -41,9 +41,9 @@ class Embedding(nn.Module): self.dropout = nn.Dropout(dropout) if not isinstance(self.embed, TokenEmbedding): - if hasattr(self, 'embed_size'): + if hasattr(self.embed, 'embed_size'): self._embed_size = self.embed.embed_size - elif hasattr(self, 'embedding_dim'): + elif hasattr(self.embed, 'embedding_dim'): self._embed_size = self.embed.embedding_dim else: self._embed_size = self.embed.weight.size(1)