whwu95 / Cap4Video

【CVPR'2023 Highlight & TPAMI】Cap4Video: What Can Auxiliary Captions Do for Text-Video Retrieval?
https://arxiv.org/abs/2301.00184
MIT License
225 stars 16 forks source link

Python, Pytorch, Torchvision, Cudatoolkit versions #22

Closed dumbelldore closed 8 months ago

dumbelldore commented 8 months ago

Unable to train the model, I think the library versions are not compatible. I'm getting the following error: unrecognized arguments: --local_rank