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 ]