Great to hear you solved it. Funny, I noticed that the ‘tolist()’ was different from what I use but due to my own short experience with PyTorch I thought ‘that’s probably just another way of doing .item()’ Lesson for me: always speak up about anything you don’t recognize…