search

RitaRamo / smallcap

SmallCap: Lightweight Image Captioning Prompted with Retrieval Augmentation
94 stars 20 forks source link

how long is "retrieving neighors" gonna take? #9

Closed nessie7purple closed 1 year ago

nessie7purple commented 1 year ago

I was reproducing the experiment accoding to README, and everything was ok until entering "retrieve captions". After images were encoded, it started to retrieve neighors. But the process seemed like frozen, nothing happened for like 2-3 hours. So I'm wondering if it is expected? How long is "retrieving neighors" gonna be?

YovaKem commented 1 year ago

Make sure you're running the retrieval on GPU. It shouldn't take more than a couple of hours.