Skip to content

Commit a5d608c

Browse files
committed
Merge branch 'dev' of github.com:fastnlp/fastNLP into dev
2 parents c72bc07 + 595fc1d commit a5d608c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fastNLP/embeddings/static_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def __init__(self, vocab: Vocabulary, model_dir_or_name: Union[str, None] = 'en'
170170
if model_path:
171171
embedding = self._load_with_vocab(model_path, vocab=lowered_vocab, init_method=init_method)
172172
else:
173-
embedding = self._randomly_init_embed(len(vocab), embedding_dim, init_method)
173+
embedding = self._randomly_init_embed(len(lowered_vocab), embedding_dim, init_method)
174174
self.register_buffer('words_to_words', torch.arange(len(vocab)).long())
175175
if lowered_vocab.unknown:
176176
unknown_idx = lowered_vocab.unknown_idx

0 commit comments

Comments
 (0)