Hi, I want to check the initial loss of my network using the following codes:
criterion = nn.MSELoss()
input = Variable(torch.rand(1,1,27,27,27))
label = Variable(torch.rand(1,1,27,27,27))
net = YLNet3D(1,9)
output = net(input)
loss = criterion(output,label)
print loss

But the loss is not what it’s supposed to be (my num_classes is 9, so should be 2.197. What might be the reasons for that? thanks!