Browse Source

更新StaticEmbedding中的bug

tags/v0.4.10
yh 6 years ago
parent
commit
d576d3999f
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      fastNLP/embeddings/static_embedding.py

+ 1
- 0
fastNLP/embeddings/static_embedding.py View File

@@ -118,6 +118,7 @@ class StaticEmbedding(TokenEmbedding):
embedding = self._load_with_vocab(model_path, vocab=lowered_vocab, init_method=init_method)
else:
embedding = self._randomly_init_embed(len(vocab), embedding_dim, init_method)
self.words_to_words = nn.Parameter(torch.arange(len(vocab)).long(), requires_grad=False)
if lowered_vocab.unknown:
unknown_idx = lowered_vocab.unknown_idx
else:


Loading…
Cancel
Save