Hi,
I found out that if I comment out this line of code:
switch_gradient([netDX, netDY], mode='off')

then everything will be fine…
But this line of code could work on v0.3.1…
Does anyone know what happened to this problem?
Thanks!!!