how to test images with the size of 512 ?

[3learning12340]

I have trained the model with images which's size is equal to 512. However, the error occured when I test the images with the size of 512. The error is shown as RuntimeError: shape '[1, 24532, 5]' is invalid for input of size 122640. File "/layers/functions/detection.py", line 38, in forward self.num_classes).transpose(2, 1). how can I solve the problem? thank you very much

[ishangupta3]

Were you able to fix this problem?