Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-02-19 18:37:17 -08:00
parent 1043832493
commit 328ad4da04
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models.py
View File

@ -140,8 +140,7 @@ class weightedFeatureFusion(nn.Module): # weighted sum of 2 or more layers http
# Adjust channels # Adjust channels
if dc > 0: # pad if dc > 0: # pad
pad = nn.ZeroPad2d((0, 0, 0, 0, 0, dc)) a = nn.ZeroPad2d((0, 0, 0, 0, 0, dc))(a)
a = pad(a)
elif dc < 0: # slice elif dc < 0: # slice
a = a[:, :nc] a = a[:, :nc]