search

cotitan / textsum-transformer

Transformer for text summarization implemented in pytorch
11 stars 0 forks source link

Which "word2vec"? #1

Open Vincent-Li-9701 opened 5 years ago

Vincent-Li-9701 commented 5 years ago

Could you please post the link of the word2vec package you used in utils.py?

cotitan commented 5 years ago

I've update the requirements in readme.md. Simply you can use pip install word2vec to install it. I use this package to load pretrained word2vec embeddings, if it's useless for you, just comment 'import word2vec'. Besides, there are packages like 'allennlp, tensorboardx, pyrouge', they are not necessary if you just use the basic transformer, you could comment the 'import xxx' statement. 'tensorboardx' is used as a pytorch-version tensorboard, if you do not use this, comment all statement contains "writer" in train.py