position embedding reshape error

[GaoXiaoshan]

Thank you for your excellent work.
When do the pretrain， there has the following error

File xxxx, line 59, in get_position_embedding
position_embedding = position_embedding[1:, :].reshape((PATCH_NUM_WIDTH, PATCH_NUM_HEIGHT, POSITION_EMBEDDING_LENGTH)).unsqueeze(0).unsqueeze(0)
RuntimeError: shape '[24, 24, 1024]' is invalid for input of size 262144

[xrorrim]

We haven't encountered this issue. Could you please provide a more detailed description of the problem?

[lucasjinreal]

Actually all uses are getting same error..