Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Fix MixConv2d() remove shortcut + apply depthwise (#5410)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-10-30 13:38:51 +02:00
1 parent 7f9bbf0268
commit 5d4258fac5
3 files changed
+13 -12
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

utils/torch_utils.py
+1 -1
View File
@@ -166,7 +166,7 @@ def initialize_weights(model):
elif t is nn.BatchNorm2d:
m.eps = 1e-3
m.momentum = 0.03
elif t in [nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6]:
elif t in [nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6, nn.SiLU]:
m.inplace = True