Skip to content

Commit 8421be6

Browse files
committed
Fixed test.
1 parent b59d5c0 commit 8421be6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/derivatives/jacobian.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,12 @@ function test_jacobian(n::Integer, T = Float32)
2828

2929
_params = params(NeuralNetwork(c, T))
3030
input = rand(T, n)
31-
@test build_nn_function(g.f, nn)(input, params) c(input, params)
32-
@test build_nn_function(derivative(g), nn)(input, params) ForwardDiff.jacobian(input -> c(input, params), input)
31+
@test build_nn_function(g.f, nn)(input, _params) c(input, _params)
32+
@test build_nn_function(derivative(g), nn)(input, _params) ForwardDiff.jacobian(input -> c(input, _params), input)
3333
end
3434

3535
for n 1:10
3636
for T (Float32, Float64)
3737
test_jacobian(n, T)
38-
end
38+
end
3939
end

0 commit comments

Comments
 (0)