Event-AHU / OpenPAR

[OpenPAR] An open-source framework for Pedestrian Attribute Recognition, based on PyTorch
MIT License
79 stars 11 forks source link

关于clip_model.encode_text中attn_mask维度不匹配 #1

Closed Keinitial closed 10 months ago

Keinitial commented 10 months ago

您好,请问您知道我这是哪里出问题了吗,提示clip_model.encode_text中attn_mask维度不匹配。self.text = clip.tokenize(attributes).to("cuda") 后的维度(attr_num, 77)。感谢您的注意! question

1125178969 commented 10 months ago

你好, 这是我们疏忽导致的一个bug,当缺少参数--use_text_prompt时会出现这个问题, 我们现在更新了代码,已经修复这个问题

Keinitial commented 10 months ago

你好, 这是我们疏忽导致的一个bug,当缺少参数--use_text_prompt时会出现这个问题, 我们现在更新了代码,已经修复这个问题

好的,感谢您的回复!