I replace the code with yours, but I got the output as this

[cuixing100876]

          I replace the code with yours, but I got the output as this![result](https://user-images.githubusercontent.com/53261018/186422373-0678c37a-8d2d-45ac-97cd-b772c99f0368.jpg)

Originally posted by @phhandong in https://github.com/diyiiyiii/StyTR-2/issues/10#issuecomment-1225680160

[cuixing100876]

I have the same problem as @phhandong

[phhandong]

I guess it's a CUDA version mismatch that's causing this problem, I solve this problem by switching my device to V100 with lower CUDA version, you can try is my way.