model fusion and onnx export

This commit is contained in:
Glenn Jocher committed 2020-06-07 13:42:33 -07:00
1 parent b810b21299
commit 12b0c046d5
3 files changed
+20 -7

No files matched your search

+3
View File
@@ -20,6 +20,9 @@ class Conv(nn.Module): # standard convolution
def forward(self, x):
return self.act(self.bn(self.conv(x)))
def fuseforward(self, x):
return self.act(self.conv(x))
class Bottleneck(nn.Module):
def __init__(self, c1, c2, shortcut=True, g=1, e=0.5): # ch_in, ch_out, shortcut, groups, expansion