We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e3bd717 commit 02dda74Copy full SHA for 02dda74
keras_cv/models/object_detection/retina_net/retina_net_test.py
@@ -210,9 +210,9 @@ def test_weights_contained_in_trainable_variables(self):
210
_ = retina_net(xs)
211
variable_names = [x.name for x in retina_net.trainable_variables]
212
# classification_head
213
- self.assertIn('RetinaNet/prediction_head/conv2d_8/kernel:0', variable_names)
+ self.assertIn("RetinaNet/prediction_head/conv2d_8/kernel:0", variable_names)
214
# box_head
215
- self.assertIn('RetinaNet/prediction_head_1/conv2d_11/kernel:0', variable_names)
+ self.assertIn("RetinaNet/prediction_head_1/conv2d_11/kernel:0", variable_names)
216
217
def test_weights_change(self):
218
bounding_box_format = "xywh"
0 commit comments