Question about reproduction

[miaoyuchun]

I am sorry to bother you. When I retrain the network (isp + mobilenet) following your advice, I got loss=nan from the first step. I didn't modify your code at all.Do you have any advice to make this code work?

[zhwhong]

I have the same problem. Have you solve it? @miaoyuchun

[cuiziteng]

I have the same issue

[zhwhong]

@cuiziteng @miaoyuchun I have this issue on NVIDIA A10 GPU before, but it works fine with RTX6000 GPU(I dont't know what exactly the reason, but it just works fine on RTX6000). Maybe you can swith to another GPU and try again. Hope it helps.

[cuiziteng]

@zhwhong Thanks !