Skip to content

GitLab

Commit 0a42034e authored by rusty1s's avatar rusty1s
Browse files

test autograd x and weight

parent 599b6042
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/test_spline_conv.py
View file @ 0a42034e
...@@ -55,10 +55,10 @@ def test_spline_weighting_backward_cpu(): ...@@ -55,10 +55,10 @@ def test_spline_weighting_backward_cpu():
op = SplineWeighting(kernel_size, is_open_spline, degree) op = SplineWeighting(kernel_size, is_open_spline, degree)
x = torch.DoubleTensor(16, 2).uniform_(-1, 1) x = torch.DoubleTensor(16, 2).uniform_(-1, 1)
x = Variable(x) x = Variable(x, requires_grad=True)
pseudo = torch.DoubleTensor(16, 3).uniform_(0, 1) pseudo = torch.DoubleTensor(16, 3).uniform_(0, 1)
pseudo = Variable(torch.DoubleTensor(pseudo), requires_grad=True) pseudo = Variable(torch.DoubleTensor(pseudo), requires_grad=True)
weight = torch.DoubleTensor(25, 2, 4).uniform_(-1, 1) weight = torch.DoubleTensor(25, 2, 4).uniform_(-1, 1)
weight = Variable(weight) weight = Variable(weight, requires_grad=True)
assert gradcheck(op, (x, pseudo, weight), eps=1e-6, atol=1e-4) is True assert gradcheck(op, (x, pseudo, weight), eps=1e-6, atol=1e-4) is True
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment