From 865e046e11783bf13fba71634b26a3260bd90c5a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 2 Jan 2021 13:04:12 -0800 Subject: [PATCH] Update yolov3-tiny.yaml --- models/yolov3-tiny.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/yolov3-tiny.yaml b/models/yolov3-tiny.yaml index 85f9fbd4..ff7638ca 100644 --- a/models/yolov3-tiny.yaml +++ b/models/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 ]