Hi,
Can you give the stack trace of where the error comes from? And what does the anomaly mode gives you as the origin for the error? (if you use nightly version of pytorch, anomaly mode should point you to all the steps in double backward).