Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "369317f45354248582884f6e2f500b7cebea2236"
Commit 09788b47 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

loss function test

parent 8e9db0f1
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
official/vision/beta/projects/yolo/losses/yolo_loss_test.py
View file @ 09788b47
...@@ -24,7 +24,11 @@ from official.vision.beta.projects.yolo.losses import yolo_loss ...@@ -24,7 +24,11 @@ from official.vision.beta.projects.yolo.losses import yolo_loss
class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
def test_loss_init(self): @parameterized.parameters(
(True),
(False),
)
def test_loss_init(self, scaled):
"""Test creation of YOLO family models.""" """Test creation of YOLO family models."""
def inpdict(input_shape, dtype = tf.float32): def inpdict(input_shape, dtype = tf.float32):
...@@ -66,7 +70,7 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): ...@@ -66,7 +70,7 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
scale_xys = {key: 2.0 for key in keys}, scale_xys = {key: 2.0 for key in keys},
max_deltas = {key: 30.0 for key in keys}, max_deltas = {key: 30.0 for key in keys},
label_smoothing=0.0, label_smoothing=0.0,
use_scaled_loss=True, use_scaled_loss=scaled,
update_on_repeat=True update_on_repeat=True
) )
......
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