Thank you very much~
And I have another question now, when I try to find the problem myself, I try to track gradient changes. But I don’t know where to find the grad_fn of each variable. So where can I find them? I mean, I want to know how they work so that I can analyze which part leading to this problem, but I can’t find them.