Code described in the paper Retrieval Augmentation for Deep Neural Networks: https://arxiv.org/abs/2102.13030
If you find our work useful in your research, please consider citing the paper:
@misc{ramos2021retrieval,
title={Retrieval Augmentation for Deep Neural Networks},
author={Rita Parada Ramos and Patrícia Pereira and Helena Moniz and Joao Paulo Carvalho and Bruno Martins},
year={2021},
eprint={2102.13030},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
Our code was adapted from:
- Image Captioning task: https://github.com/sgrvinod/a-PyTorch-Tutorial-to-Image-Captioning
- Sentiment Analysis task: https://github.com/bentrevett/pytorch-sentiment-analysis/blob/master/2%20-%20Upgraded%20Sentiment%20Analysis.ipynb