albanD:

Do you have custom autograd Function in your code?


No i don’t have this Alban;

Enabling anomaly mode might help you pinpoint the faulty Function.

Can you share how do we enable that? (Will post back what i found!)
Thanks;