From a01e2ade25f8a075606adda768e048e881d0d311 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Apr 2019 11:18:51 -0400 Subject: [PATCH] Update embeddings.py --- pytorch_lightning/utils/embeddings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/utils/embeddings.py b/pytorch_lightning/utils/embeddings.py index 5f064fcb..507848b2 100644 --- a/pytorch_lightning/utils/embeddings.py +++ b/pytorch_lightning/utils/embeddings.py @@ -97,11 +97,11 @@ class PretrainedEmbedding(torch.nn.Embedding): if __name__ == '__main__': emb = PretrainedEmbedding( - embedding_path='/Users/waf/Developer/NGV/research-fermat/fermat/.vector_cache/glove.840B.300d.txt', + embedding_path='/Users/waf/Developer', embedding_dim=300, task_vocab={'hello': 1, 'world': 2} ) data = torch.Tensor([[0, 1], [0, 2]]).long() embedded = emb(data) - print(embedded) \ No newline at end of file + print(embedded)