Internal change

PiperOrigin-RevId: 534299624

Internal change
PiperOrigin-RevId: 534299624
02bbd23f · Jiageng Zhang · A. Unique TensorFlower · 1dba9168 · 02bbd23f
显示空白变更内容
内联并排

Showing with 4 addition and 1 deletion

official/projects/yolo/modeling/backbones/darknet_test.py official/projects/yolo/modeling/backbones/darknet_test.py +4 -1

未找到文件。
--- 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)