Hi @Andras7
I reorganized your code a little bit to make it easily installable with pip.. You can install it with:
pip install git+https://github.com/marta-sd/word2vec-pytorch.git
You can take a look at my version here
If you are interested in this kind of contribution I'd be happy to create a PR
Best
MSD