We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5b0cdcc commit 1db3967Copy full SHA for 1db3967
tests/test_ops/test_conv_gradfix.py
@@ -37,6 +37,7 @@ def test_conv2d_transposed_cuda(self):
37
weight = self.weight.cuda()
38
res = conv_transpose2d(x, weight, None, 1, 1)
39
assert res.shape == (1, 1, 32, 32)
40
- gradcheck(conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)
+ gradcheck(
41
+ conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)
42
gradgradcheck(
43
conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)
0 commit comments