diff --git a/opacus/tests/grad_samples/conv3d_test.py b/opacus/tests/grad_samples/conv3d_test.py index a29097ab..afa01b4b 100644 --- a/opacus/tests/grad_samples/conv3d_test.py +++ b/opacus/tests/grad_samples/conv3d_test.py @@ -71,7 +71,7 @@ def test_conv3d( groups=groups, ) is_ew_compatible = ( - dilation == 1 or padding != "same" + dilation == 1 and padding != "same" ) # TODO add support for padding = 'same' with EW self.run_test( x,