@ptrblck … Thanks for the clarification i missed… but can you tell me here why the error still exists:
loss = criterion(output, target)
although output shape and target are the same (torch.Size([16, 1]))