This commit is contained in:
Giovanni Campagna 2019-03-13 16:34:12 -07:00
parent 919768b42b
commit f65d939f9a
1 changed files with 3 additions and 1 deletions

View File

@ -41,6 +41,8 @@ ENTITIES = ['DATE', 'DURATION', 'EMAIL_ADDRESS', 'HASHTAG',
MAX_ARG_VALUES = 5
class AlmondEmbeddings(torchtext.vocab.Vectors):
def __init__(self, name=None, cache=None, **kw):
super().__init__(name, cache, **kw)
def cache(self, name, cache, url=None):
del name
@ -60,7 +62,7 @@ class AlmondEmbeddings(torchtext.vocab.Vectors):
self.itos = itos
self.stoi = {word: i for i, word in enumerate(itos)}
self.vectors = torch.Tensor(vectors).view(-1, dim)
self.vectors = torch.stack(vectors, dim=0).view(-1, dim)
self.dim = dim