I think there’s something wrong happening in your model because your 32 examples that should be ran in parallel disappears into a single example as the output. The output should be torch.size([32, 1]). Could you show how net looks like?