RuntimeError: shape '[1, 5, 7, 5, 7, 1]' is invalid for input of size 1369

maggiez0138 / Swin-Transformer-TensorRT

This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.

MIT License

161 stars 29 forks source link

Open iPengXPro opened 1 year ago

iPengXPro commented 1 year ago

Hi! When I change the default img_size 224 to 150,I got the error.Is there any other parameters i need to change to?