Skip to content
Snippets Groups Projects
Commit 7c75fde3 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed bug

parent e37444e7
No related branches found
No related tags found
2 merge requests!490.3.1,!40Add ConvDepthWise support
Pipeline #71614 passed
......@@ -237,7 +237,7 @@ void convolution_depthwise_forward(
const int iOffset = NB_CHANNELS * iPos;
const int wOffset = (output*KERNEL_HEIGHT + sy)
* KERNEL_WIDTH * NB_CHANNELS;
* KERNEL_WIDTH;
if (DILATION_X == 1 && ((PADDING_X == 0
&& OUTPUTS_WIDTH == OUTPUTS_WIDTH_NOPAD)
......
......@@ -411,7 +411,7 @@ class test_operator_export(unittest.TestCase):
self.unit_test_export(model, "Conv2D", [[1, 3, 12, 12]], False, False)
def test_export_convDepthWise2D(self):
print("Conv2D")
print("ConvDepthWise2D")
model = aidge_core.sequential([
aidge_core.ConvDepthWise2D(nb_channels=3, kernel_dims=(3, 3), name="conv")
])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment