The shape of the output of the network is [batch_size, 2] exactly. But the question is that the value of the output is not usually in the range [0, 1], which confused me a lot. For example, the output can get a strange value such as [9.6655, -8.4452]. Does it mean that my model architecture is wrong?
In addition, today I set the batch_size 2 and add a BN Layer behind the FC Layer. It makes the value of train loss does not keep 0 or 0.69 anymore. Instead, the absolute value of the output can be in the range [0, 1] with the existence of minus . What’s more, the train loss still has not a trend of decline.