Update embeddings.py
This commit is contained in:
parent
3e9f37a382
commit
a01e2ade25
|
@ -97,7 +97,7 @@ class PretrainedEmbedding(torch.nn.Embedding):
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
emb = PretrainedEmbedding(
|
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,
|
embedding_dim=300,
|
||||||
task_vocab={'hello': 1, 'world': 2}
|
task_vocab={'hello': 1, 'world': 2}
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue