The outputs depend on the activation function of the output layer in your model. You must be careful when placing this activation function because each type of loss function has its own range of values.
In your code snip, if you use Cross-Entropy Loss, you should not use any activation function in the output layer. However, you took argmax in the channel dimension of your ground-truth which has only 1 channel, this step does not make sense.