Unexpected keyword argument 'batch_first'

anosorae / IRRA

Cross-Modal Implicit Relation Reasoning and Aligning for Text-to-Image Person Retrieval (CVPR 2023)

MIT License

194 stars 27 forks source link

Unexpected keyword argument 'batch_first' #4

Closed LKRCharon closed 1 year ago

LKRCharon commented 1 year ago

$ bash run_irra.sh

File "./IRRA-main/model/build.py", line 148, in build_model
    model = IRRA(args, num_classes)
  File "./IRRA-main/model/build.py", line 27, in __init__
    self.cross_attn = nn.MultiheadAttention(self.embed_dim,
TypeError: __init__() got an unexpected keyword argument 'batch_first'

anosorae commented 1 year ago

This is a Pytorch version issue, please upgrade your Pytorch to >= 1.9.0.