Skip to content

Commit 595fc1d

Browse files
authored
mirror fix (#361)
1 parent 0f4dad1 commit 595fc1d

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)