Results 2 comments of Seth Claudepierre

I just encountered this as well. On a hunch, I downgraded `torch` from v2.5 to v2.4 and the `nan` values are gone. Perhaps check the version of `torch` that you...