Skip to content

Commit

Permalink
Fix lint (#2702)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhouzaida authored Mar 25, 2023
1 parent 5b0cdcc commit 1db3967
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/test_ops/test_conv_gradfix.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def test_conv2d_transposed_cuda(self):
weight = self.weight.cuda()
res = conv_transpose2d(x, weight, None, 1, 1)
assert res.shape == (1, 1, 32, 32)
gradcheck(conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)
gradcheck(
conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)
gradgradcheck(
conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)

0 comments on commit 1db3967

Please sign in to comment.