search

Haiyang-W / DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"
https://arxiv.org/abs/2301.06051
Apache License 2.0
361 stars 28 forks source link

Question about pytorch version #15

Closed sky-fly97 closed 1 year ago

sky-fly97 commented 1 year ago

Hello, my pytorch version is 1.8.1, I encountered this problem while training, does the pytorch version have to be higher than 1.9.0? ../pcdet/models/backbones_3d/dsvt.py", line 212, in __init__ self.self_attn = nn.MultiheadAttention(d_model, nhead, dropout=dropout, batch_first=batch_first) TypeError: __init__() got an unexpected keyword argument 'batch_first'

Haiyang-W commented 1 year ago

yes, please try Pytorch 1.10. MultiheadAttention may be different between 1.8 and 1.10.