test_broken AlphaDropout

2020-02-16 21:22:37 +05:30 · 2020-02-16 21:22:37 +05:30 · 26631e1361
parent 29ab410794
commit 26631e1361
1 changed files with 1 additions and 1 deletions
--- a/test/cuda/layers.jl
+++ b/test/cuda/layers.jl
@ -15,7 +15,7 @@ function gradtest(name::String, layers::Vector, xs = nothing, args...)
      @testset "$layer GPU grad test" begin
        l = gpu(layer(args...))
        xs = gpu(xs)
-        if l isa DepthwiseConv
+        if l isa DepthwiseConv || l isa AlphaDropout
          ps = Flux.params(l)
          @test_broken gradient(() -> sum(l(xs)), ps) isa Flux.Zygote.Grads
        else