muzairkhattak/ViFi-CLIP

CUDA out of memory

Opened this issue · 1 comments

W4ngH4o commented

Hi,

I am trying to reproduce the reported results when training on 4 2080ti GPUs. But even if I set batchsize=4,frames=2 in Zero-shot setting, it ends CUDA out of memory. May I ask if you have any suggestions?

Thank you!

muzairkhattak commented

Hi @W4ngH4o,

Thank you for showing interest in ViFi-CLIP!

It is strange that it is providing CUDA out of memory error.

Can you let us know if you are using the same PyTorch version that is mentioned in our environment requirements file?