This commit is contained in:
mehrad 2019-05-29 16:42:36 -07:00
parent a27c3a8d8b
commit 4004664259
2 changed files with 2 additions and 1 deletions

View File

@ -30,6 +30,7 @@
from subprocess import Popen, PIPE
import logging
import re
import os
import string
import numpy as np
import collections

View File

@ -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: