Thank you, the explanation of what logits are helps. So my normalization values before training is nothing I should worry about either? Or should I keep just ToTensor()?
And indeed, my last outputs of my model is:
(3): ReLU()
(4): Conv2d(256, 1, kernel_size=(1, 1), stride=(1, 1))