The parameter batch_first=True is causing an error

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

The parameter batch_first=True is causing an error #16

Closed Tiiivoo closed 9 months ago

Tiiivoo commented 9 months ago

When I run the code with batch_first=True, I encounter an error. Upon researching, it seems that this parameter is only available in Torch 1.9.0. How can I resolve this?