search

YiLunLee / missing_aware_prompts

Multimodal Prompting with Missing Modalities for Visual Recognition, CVPR'23
https://yilunlee.github.io/missing_aware_prompts/
160 stars 9 forks source link

a training problem #16

Closed ZunxiaoXu closed 1 year ago

ZunxiaoXu commented 1 year ago

hi, thanks for your great work! I try to train the model based on the hateful dataset, but when the first epoch reaches 34%, the training process will always get stuck here. I don't know why. Have you encountered this situation?

ZunxiaoXu commented 1 year ago

I solved this problem by setting only train with a single gpu

ALEX13679173326 commented 10 months ago

I also encountered the same problem. Using one GPU is too slow. Is there a solution for running on multiple GPUs?