Hello, thank you for reply, iam using this type of loss- MSE
loss = ((y - y_pred) ** 2).mean()

Please can you give me example how to use “dice loss” ? I was searching on internet but I didn’t get any solution for this model.