search

muzairkhattak / ViFi-CLIP

[CVPR 2023] Official repository of paper titled "Fine-tuned CLIP models are efficient video learners".
https://muzairkhattak.github.io/ViFi-CLIP/
MIT License
248 stars 18 forks source link

CUDA out of memory #16

Open W4ngH4o opened 10 months ago

W4ngH4o commented 10 months ago

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 10 months ago

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?