Hi,
Yes we added more checks for operations that were potentially invalid.
Could you run your code with anomaly mode 8 by setting autograd.set_detect_anomaly(True) at the beginning of your script. That will let us know what is causing the error.