retrieval code of Tag2text

Thanks for your nice work!
I have read through your good paper and code. It's an excellent work.

However, I didn't find any code about the retrieval-related codes. In your ICLR2024 paper, there are three pertaining objectives, but there are only two losses in your code. Did I miss something?
Also the retrieval inference codes.

Looking forward to your reply. 