fixed depthwise conv produce filtersize=None

vangjiwei · Oct 20, 2020 · d973969 · d973969
1 parent 7e85a91
commit d973969
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/gen_config.py b/scripts/gen_config.py
@@ -144,7 +144,7 @@ def gen_conv2d_config(layer, output_shifts, bias_shifts):
     c = c.replace('<bias>',convert_tensor_name(layer.weights[1]))
     c = c.replace('<output_shift_values>', output_shifts)
     c = c.replace('<bias_shift_values>', bias_shifts)
-    c = c.replace('<filter_size>', str(layer.filters)) # output channel
+    c = c.replace('<filter_size>', '1' if layer.filters is None else str(layer.filters))  # output channel
     c = c.replace('<kernel_size>', to_cstyle(layer.kernel_size))
     c = c.replace('<stride_size>', to_cstyle(layer.strides))
     c = c.replace('<padding_size>', '{0, 0}') # not using it with keras, defined by padding type instead