search

BigRedT / info-ground

Learning phrase grounding from captioned images through InfoNCE bound on mutual information
http://tanmaygupta.info/info-ground/
Other
72 stars 16 forks source link

GPU memory #4

Open OliverHuang1220 opened 3 years ago

OliverHuang1220 commented 3 years ago

Hey,@BigRedT Thanks for your excellent job.Can you tell me the code will use how many GPU memory? I have only one NVIDIA TITAN XP 12GB.