From 0d568f6faf7fe7263616a033e39dcdcd4f867584 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Tue, 17 Oct 2017 21:33:11 +0100 Subject: [PATCH] fix tests --- test/tracker.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/tracker.jl b/test/tracker.jl index dca7e13c..48b45f09 100644 --- a/test/tracker.jl +++ b/test/tracker.jl @@ -11,8 +11,8 @@ gradtest(f, dims...) = gradtest(f, rand.(dims)...) @test gradtest(x -> sin.(sum(x, (2, 3))), (3,4,5)) -@test gradtest(x -> softmax(x).*(1:3), 3) -@test gradtest(x -> softmax(x).*(1:3), (3,5)) +@test gradtest(x -> NNlib.softmax(x).*(1:3), 3) +@test gradtest(x -> NNlib.softmax(x).*(1:3), (3,5)) @test gradtest(Flux.mse, rand(5,5), rand(5, 5)) @test gradtest(Flux.crossentropy, rand(5,5), rand(5, 5))