diff --git a/models/tf.py b/models/tf.py
index b70e37488009..3428032a0aa7 100644
--- a/models/tf.py
+++ b/models/tf.py
@@ -88,10 +88,10 @@ def call(self, inputs):
 
 class TFDWConv(keras.layers.Layer):
     # Depthwise convolution
-    def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True, w=None):
+    def __init__(self, c1, c2, k=1, s=1, p=None, act=True, w=None):
         # ch_in, ch_out, weights, kernel, stride, padding, groups
         super().__init__()
-        assert g == c1 == c2, f'TFDWConv() groups={g} must equal input={c1} and output={c2} channels'
+        assert c1 == c2, f'TFDWConv() input={c1} must equal output={c2} channels'
         conv = keras.layers.DepthwiseConv2D(
             kernel_size=k,
             strides=s,