I am not storing this error, it will be returned to def train() and their i am simply printing it.
If i set my error=dice_loss() then  the model is working fine. But i need the other loss also, which is a BCELoss.