Hi all,
After drilling deep down to replicating the DataLoader’s tensor collation I’ve actually found out that the cause for my error was indeed in the batch’s target tensors. Some of the y tensors have the shape [32, 32] while others have [32, 33]. This wasn’t immediately clear, as the shuffle brings out these tensors up by chance. Thus I rest my case and go back fixing my code.
For me this problem is thus resolved for now.
Sorry for unnecessary call-outs!