使用 ExternalAttention 出现错误
作者: yy-wy创建于 2021年7月2日更新于 2025年12月29日
有人在使用时遇到过这个错误吗？ RuntimeError: mat1 dim 1 must match mat2 dim 0 nn.Linear 应该没有问题 mk: Linear(in_features=576, out_features=8, bias=False) mv: Linear(in_features=8, out_features=576, bias=False) 另外主干程序使用 EAattention 没有问题
内容来源: xmu-xiaoma666/External-Attention-pytorch