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)