can't use with pytorch dataloader
RuntimeError: einsum(): the number of subscripts in the equation (3) does not match the number of dimensions (4) for operand 0 and no ellipsis was given
Please paste the whole code and describe your task in detail<|||||>@sipie800 Seems like a dimension error while resizing since you are using einsum. If you could share your code I maybe able to help.<|||||>This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/master/CONTRIBUTING.md) are likely to be ignored.