Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 564420337
  • Loading branch information
fyangf authored and tensorflower-gardener committed Sep 11, 2023
1 parent 846f356 commit c6bd774
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions official/vision/modeling/retinanet_model_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,37 +36,37 @@ class RetinaNetTest(parameterized.TestCase, tf.test.TestCase):
'use_separable_conv': True,
'build_anchor_boxes': True,
'is_training': False,
'has_att_heads': False
'has_att_heads': False,
},
{
'use_separable_conv': False,
'build_anchor_boxes': True,
'is_training': False,
'has_att_heads': False
'has_att_heads': False,
},
{
'use_separable_conv': False,
'build_anchor_boxes': False,
'is_training': False,
'has_att_heads': False
'has_att_heads': False,
},
{
'use_separable_conv': False,
'build_anchor_boxes': False,
'is_training': True,
'has_att_heads': False
'has_att_heads': False,
},
{
'use_separable_conv': False,
'build_anchor_boxes': True,
'is_training': True,
'has_att_heads': True
'has_att_heads': True,
},
{
'use_separable_conv': False,
'build_anchor_boxes': True,
'is_training': False,
'has_att_heads': True
'has_att_heads': True,
},
)
def test_build_model(self, use_separable_conv, build_anchor_boxes,
Expand Down

0 comments on commit c6bd774

Please sign in to comment.