Open SeaLu opened 5 months ago
When I val the model, the error is : mid_out=outer_tokens.reshape(b,int(math.sqrt(l)),int(math.sqrt(l)),m).permute(0,3,1,2) RuntimeError: shape '[32, 75, 75, 16]' is invalid for input of size 2924544
Hi,I can help you with that, would you mind telling me your email?
I input 512×512 images into the network and no errors are reported.I think reporting this error has something to do with the size of the image. @SeaLu
When I input 640*640 images to train the backbone of the model, the error is
![erroe](https://github.com/txchen-USTC/MiM-ISTD/assets/78088515/0ee10d3d-85a4-4867-ab81-e941d97108af)