fixes
This commit is contained in:
parent
a27c3a8d8b
commit
4004664259
|
@ -30,6 +30,7 @@
|
|||
from subprocess import Popen, PIPE
|
||||
import logging
|
||||
import re
|
||||
import os
|
||||
import string
|
||||
import numpy as np
|
||||
import collections
|
||||
|
|
|
@ -70,7 +70,7 @@ def load_embeddings(args, logger=_logger):
|
|||
logger.info(f'Getting pretrained word vectors')
|
||||
final_vectors = []
|
||||
if args.use_fastText:
|
||||
vectors = torchtext.vocab.FastText(cache=args.embeddings, language='fa')
|
||||
vectors = [torchtext.vocab.FastText(cache=args.embeddings, language='fa')]
|
||||
else:
|
||||
char_vectors = torchtext.vocab.CharNGram(cache=args.embeddings)
|
||||
if args.small_glove:
|
||||
|
|
Loading…
Reference in New Issue