"model/vscode:/vscode.git/clone" did not exist on "3b96a93672377129f2a2aafc447e79ef1ca48c5f"
Commit 43d232e5 authored by Jaehong Kim's avatar Jaehong Kim Committed by A. Unique TensorFlower
Browse files

Remove duplicated bias and activation from the quantized separable conv.

PiperOrigin-RevId: 446001473
parent 2aff7786
......@@ -64,6 +64,10 @@ class SeparableConv2DQuantized(tf.keras.layers.Layer):
# Depthwise conv input filters is always equal to output filters.
# This filters argument only needed for the point-wise conv2d op.
del dwconv_kwargs['filters']
dwconv_kwargs.update({
'activation': None,
'use_bias': False,
})
self.dw_conv = depthwise_conv2d_quantized(name='dw', **dwconv_kwargs)
conv_kwargs = self._conv_kwargs.copy()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment