RuntimeError: shape '[-1, 2, 768]' is invalid for input of size 33024

TencentYoutuResearch / PersonReID-NAFS

Code for "Contextual Non-Local Alignment over Full-Scale Representation for Text-Based Person Search"

https://arxiv.org/abs/2101.03036

Other

61 stars 10 forks source link

RuntimeError: shape '[-1, 2, 768]' is invalid for input of size 33024 #18

Open liuliu408 opened 3 years ago

liuliu408 commented 3 years ago

Please, what is the reason for this? File "/home/bobo/E/xxx/paper4_reid_2021/PersonReID-NAFS-main/models/model.py", line 76, in forward local_text_feat = local_text_feat.view(-1, n_sep, local_text_feat.size(1)) RuntimeError: shape '[-1, 2, 768]' is invalid for input of size 33024

Video-AD commented 2 years ago

Please, what is the reason for this? File "/home/bobo/E/xxx/paper4_reid_2021/PersonReID-NAFS-main/models/model.py", line 76, in forward local_text_feat = local_text_feat.view(-1, n_sep, local_text_feat.size(1)) RuntimeError: shape '[-1, 2, 768]' is invalid for input of size 33024

Hi, I have also encountered this problem, have you solved it? Thank you!