Honee-W/CPTNN

TRANSFORMER.py

Mohammadhakimkhah opened this issue · 1 comments

Mohammadhakimkhah commented

Hi , Thank you for your reply
in TRANSFORMER.py line94 " local_feat = th.reshape(x.permute(0, 2, 1, 3), (-1, L, C)) "
In (0, 2, 1, 3), you should not put (0, 2, 3, 1) instead of (0, 2, 1, 3) ?

Honee-W commented

Thanks for your question, i do find this bug however i forget to update the code. Now it has been fixed.