-
Notifications
You must be signed in to change notification settings - Fork 11
Open
Description
Hi,
How did you run the SwitT results? When loading the architecture and inferencing an image I get the following error:
einops.EinopsError: Error while processing rearrange-reduction pattern "b (nw_h w_h) (nw_w w_w) (h d) -> b h (nw_h nw_w) (w_h w_w) d". Input tensor shape: torch.Size([256, 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
Metadata
Metadata
Assignees
Labels
No labels