Hi, I have a question. If "x.shape[1] == L+self.num_mem_token", then it seems that the i in "mode: x[:, i, :]" does not correspond to the position of mode, because the first position is mem _token. about actionmae HOT 8 CLOSED

Hello?

from actionmae.

Can you elaborate on your question a bit more?

from actionmae.

Should the code (else: x = x[:, self.num_mem_token:, :])be added here(https://github.com/sangminwoo/ActionMAE/blob/4ae5a813ebf0ea9f4ba3655d6cefb215642309c4/lib/modeling/actionmae.py#LL363C1-L363C1)?

from actionmae.

In L359, if "x.shape[1] != L+self.num_mem_token" is False, in which line did you see the error?

from actionmae.

In L359, if "x.shape[1] ! = L+self.num_mem_token" is false, then L226 will not be executed and the cls token will be preserved, resulting in errors in L365-368 (the indexes seem to be out of correspondence).

from actionmae.

Hi?

from actionmae.

Sorry for the late reply.
I acknowledge your point. Please try the fix, and report back if you see any further errors.
Many thanks.

from actionmae.

Thank you very much for your answer.

from actionmae.