^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\luoxu.conda\envs\py312\Lib\site-packages\torch\nn\modules\module.py", line 1784, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\luoxu.conda\envs\py312\Lib\site-packages\fireredasr\models\module\conformer_encoder.py", line 67, in forward
out = self.mhsa(out, out, out, pos_emb, mask=slf_attn_mask)[0]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\luoxu.conda\envs\py312\Lib\site-packages\torch\nn\modules\module.py", line 1773, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\luoxu.conda\envs\py312\Lib\site-packages\torch\nn\modules\module.py", line 1784, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\luoxu.conda\envs\py312\Lib\site-packages\fireredasr\models\module\conformer_encoder.py", line 316, in forward
attn_scores = matrix_ac + matrix_bd
~~~~~~~~~~^~~~~~~~~~~
RuntimeError: The size of tensor a (10682) must match the size of tensor b (2842) at non-singleton dimension 3