UNITER icon indicating copy to clipboard operation
UNITER copied to clipboard

Retrieval time

Open antoine77340 opened this issue 3 years ago • 1 comments

Thank you so much for providing the code for UNITER, great work! The README mentions the following "NOTE: Image-Text Retrieval is computationally heavy, especially on COCO."

Do you have a rough estimation in mind of how many hours (or days) does it take to run the full text-to-image retrieval on COCO on 1 GPU (eg. Tesla V100).

Thanks for your help!

antoine77340 avatar Oct 28 '20 17:10 antoine77340

The size of val and test set on COCO is very large, conduct one evaluation with 8 GPUs (V100) needs 2~3hours. If you only evaluate on val and test set once after training, it will take more than 8 hours with 8 GPUs.

RenShuhuai-Andy avatar Dec 22 '20 02:12 RenShuhuai-Andy