search

whwu95 / Text4Vis

【AAAI'2023 & IJCV】Transferring Vision-Language Models for Visual Recognition: A Classifier Perspective
MIT License
204 stars 15 forks source link

Training and testing on single dGPU #1

Closed bashimr closed 1 year ago

bashimr commented 1 year ago

Hi Guys

Looks like a great project. Nice work. I am trying to explore it for zero-shot. I have an NVIDIA GeForce RTX 3080 on my machine. The instructions are to run the training on single machine with 8 GPUs. Is it possible/worthwhile to train and test the model on my machine? If so, any pointers to do that will be much appreciated.

Thanks.

Imran

whwu95 commented 1 year ago

Thank you for your interest in our work. You can perform zero-shot evaluation for a single card by simply changing --nproc_per_node=8 to --nproc_per_node=1. But I think it might be difficult to train video models with a single card.

bashimr commented 1 year ago

Thank you @whwu95.