word2vec-pytorch
word2vec-pytorch copied to clipboard
fixed issue of passing empty parameter list to optimizer
Hi @Andras7, first I want to thank you for providing this code, it really is a big help. I ran into this error when trying to train the model:
Traceback (most recent call last):
File "/path/to/test.py", line 8, in
So I went ahead and wrapped self.skip_gram_model.parameters() in list(). Might this be a version issue? I ran it with python=3.8, torch=1.7.1
Works for me now, hope this is fine with you? Feel free to accept the PR.