Skip to content

Commit

Permalink
Fix Tf export for BottleneckCSP (#7330)
Browse files Browse the repository at this point in the history
  • Loading branch information
nrupatunga authored Apr 7, 2022
1 parent 0ca85ed commit b7faeda
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/tf.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5, w=None):
self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
self.bn = TFBN(w.bn)
self.act = lambda x: keras.activations.relu(x, alpha=0.1)
self.act = lambda x: keras.activations.swish(x)
self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)])

def call(self, inputs):
Expand Down

0 comments on commit b7faeda

Please sign in to comment.