Skip to content

GitLab

Commit b1f77678 authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

List WeightedDiceClassificationLoss at top of docstring. 

PiperOrigin-RevId: 406929565
parent ab70b0f2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
research/object_detection/core/losses.py
View file @ b1f77678
...@@ -25,6 +25,7 @@ Classification losses: ...@@ -25,6 +25,7 @@ Classification losses:
* WeightedSoftmaxClassificationLoss * WeightedSoftmaxClassificationLoss
* WeightedSoftmaxClassificationAgainstLogitsLoss * WeightedSoftmaxClassificationAgainstLogitsLoss
* BootstrappedSigmoidClassificationLoss * BootstrappedSigmoidClassificationLoss
* WeightedDiceClassificationLoss
""" """
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import division from __future__ import division
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment