From 0b6266f5e0eab11218871d5560bf9b93f7547aac Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 2 Jan 2021 13:01:05 -0800 Subject: [PATCH] Update yolov3-tiny.yaml (#1823) --- models/hub/yolov3-tiny.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/hub/yolov3-tiny.yaml b/models/hub/yolov3-tiny.yaml index 85f9fbd..ff7638c 100644 --- a/models/hub/yolov3-tiny.yaml +++ b/models/hub/yolov3-tiny.yaml @@ -22,7 +22,7 @@ backbone: [-1, 1, Conv, [256, 3, 1]], [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 9-P5/32 [-1, 1, Conv, [512, 3, 1]], - [-1, 1, nn.ZeroPad2d, [0, 1, 0, 1]], # 11 + [-1, 1, nn.ZeroPad2d, [[0, 1, 0, 1]]], # 11 [-1, 1, nn.MaxPool2d, [2, 1, 0]], # 12 ]