Hello, author, I want to know how to test this code in other Dataset? Shall I retrain the code? Or Just change the image transform?