Skip to content

GitLab

Commit 6de77679 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

addressing comments

parent 73e9d6dd
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
View file @ 6de77679
...@@ -51,7 +51,6 @@ class YoloLayer(tf.keras.Model): ...@@ -51,7 +51,6 @@ class YoloLayer(tf.keras.Model):
max_delta: gradient clipping to apply to the box loss. max_delta: gradient clipping to apply to the box loss.
loss_type: `str` for the typeof iou loss to use with in {ciou, diou, loss_type: `str` for the typeof iou loss to use with in {ciou, diou,
giou, iou}. giou, iou}.
use_tie_breaker: TODO unused?
iou_normalizer: `float` for how much to scale the loss on the IOU or the iou_normalizer: `float` for how much to scale the loss on the IOU or the
boxes. boxes.
cls_normalizer: `float` for how much to scale the loss on the classes. cls_normalizer: `float` for how much to scale the loss on the classes.
......
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