apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update yolo.py nn.zeroPad2d() (#1638)

Browse Source
This commit is contained in:
Glenn Jocher 2021-01-03 11:42:10 -08:00 committed by GitHub
parent 865e046e11
commit 7d9535f80e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/yolo.py
View File

@ -240,9 +240,6 @@ def parse_model(d, ch): # model_dict, input_channels(3)
n = 1 n = 1
elif m is nn.BatchNorm2d: elif m is nn.BatchNorm2d:
args = [ch[f]] args = [ch[f]]
elif m is nn.ZeroPad2d:
args = [args]
c2 = ch[f]
elif m is Concat: elif m is Concat:
c2 = sum([ch[-1 if x == -1 else x + 1] for x in f]) c2 = sum([ch[-1 if x == -1 else x + 1] for x in f])
elif m is Detect: elif m is Detect: