word list size - Githubissues

ruotianluo / self-critical.pytorch

Unofficial pytorch implementation for Self-critical Sequence Training for Image Captioning. and others.

MIT License

996 stars 279 forks source link

word list size #209

Open aaronma2020 opened 4 years ago

aaronma2020 commented 4 years ago

hi, @ruotianluo in scripts/prepro_labels.py you use all captions When build a word-to-id dictionary. I think it is more appropriate to use training captions only. That's right?

ruotianluo commented 4 years ago

agree