From 02bbd23f37fc85bbc4ad0540a08c383dddc9b501 Mon Sep 17 00:00:00 2001 From: Jiageng Zhang Date: Mon, 22 May 2023 22:37:46 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 534299624 --- official/projects/yolo/modeling/backbones/darknet_test.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/official/projects/yolo/modeling/backbones/darknet_test.py b/official/projects/yolo/modeling/backbones/darknet_test.py index 7f6e5798e..e775f78a7 100644 --- a/official/projects/yolo/modeling/backbones/darknet_test.py +++ b/official/projects/yolo/modeling/backbones/darknet_test.py @@ -82,7 +82,10 @@ class DarknetTest(parameterized.TestCase, tf.test.TestCase): with strategy.scope(): network = darknet.Darknet( - model_id='darknet53', min_size=3, max_size=5, use_sync_bn=use_sync_bn + model_id='darknet53', + min_level=3, + max_level=5, + use_sync_bn=use_sync_bn, ) _ = network(inputs) -- GitLab