The Ground Truth label is [0.0,1.0]. And I want to predict this scalar value.
In this case, What losses can be used ?
Can I use MSE loss with nn.Linear(input_dim, 1)?
networks = nn.Sequential(... , nn.Linear(input_dim, 1)
predictions = networks(input)
loss = MSE(predictions, GT)

Thanks  in advance !