From 7d9535f80efe6fe7b78dfc5da2f3d9fa6e606a4f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 3 Jan 2021 11:42:10 -0800 Subject: [PATCH] Update yolo.py nn.zeroPad2d() (#1638) --- models/yolo.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/models/yolo.py b/models/yolo.py index 191bd77c..7ef9d501 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -240,9 +240,6 @@ def parse_model(d, ch): # model_dict, input_channels(3) n = 1 elif m is nn.BatchNorm2d: args = [ch[f]] - elif m is nn.ZeroPad2d: - args = [args] - c2 = ch[f] elif m is Concat: c2 = sum([ch[-1 if x == -1 else x + 1] for x in f]) elif m is Detect: