Hello, I used the smaller copy set "SimSIN-simple" provided by you for training, and then I ran "demo.py" with the obtained weights, hoping to get some expected results, but the results were not good, I felt completely ineffective, and the pictures I got were all purple, may I ask why I did not modify your code

[iSuryxin]

[2579690686]

I'm having the same issue, do you have any results?

[2579690686]

If you look at the loss during training, I think the loss must not converge

[choyingw]

see #25