search

diyiiyiii / StyTR-2

StyTr2 : Image Style Transfer with Transformers
352 stars 64 forks source link

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) #31

Open cuixing100876 opened 8 months ago

cuixing100876 commented 8 months ago 
          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 commented 8 months ago

I have the same problem as @phhandong

phhandong commented 8 months ago

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.