Results 3 comments of Ivan Sedykh

just put .long() for second argument in place where loss computed idk but this worked

Sure, I hope to prepare PR later this week.

bump The issue still persists, I will work on this later