复现问题

[acceleratelch]

Swin Transformer复现时出现如下错误:
Input tensor shape: torch.Size([1, 8, 8, 96]). Additional info: {'h': 3, 'w_h': 7, 'w_w': 7}.
Shape mismatch, can't divide axis of length 8 in chunks of 7
einops函数问题，以前复现好像还没这个错误，有人遇到一样的问题了吗。