LossDice: add comment

This commit is contained in:
Starbeamrainbowlabs 2022-12-12 18:34:20 +00:00
parent 449bc425a7
commit 4e4d42a281
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -35,6 +35,7 @@ def dice_coef_loss(y_true, y_pred, **kwargs):
class LossDice(tf.keras.losses.Loss):
"""An implementation of the dice loss function.
@source
Args:
smooth (float): The batch size (currently unused).
"""