Skip to content

Commit 02dda74

Browse files
committed
trainable variables fix
1 parent e3bd717 commit 02dda74

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras_cv/models/object_detection/retina_net/retina_net_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -210,9 +210,9 @@ def test_weights_contained_in_trainable_variables(self):
210210
_ = retina_net(xs)
211211
variable_names = [x.name for x in retina_net.trainable_variables]
212212
# classification_head
213-
self.assertIn('RetinaNet/prediction_head/conv2d_8/kernel:0', variable_names)
213+
self.assertIn("RetinaNet/prediction_head/conv2d_8/kernel:0", variable_names)
214214
# box_head
215-
self.assertIn('RetinaNet/prediction_head_1/conv2d_11/kernel:0', variable_names)
215+
self.assertIn("RetinaNet/prediction_head_1/conv2d_11/kernel:0", variable_names)
216216

217217
def test_weights_change(self):
218218
bounding_box_format = "xywh"

0 commit comments

Comments
 (0)