disable rnn tests
This commit is contained in:
parent
ac467cfe77
commit
bc33108e66
|
@ -41,5 +41,5 @@ end
|
|||
if CuArrays.libcudnn != nothing
|
||||
@info "Testing Flux/CUDNN"
|
||||
include("cudnn.jl")
|
||||
include("curnn.jl")
|
||||
# include("curnn.jl")
|
||||
end
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
using Flux, CuArrays, Test
|
||||
|
||||
@testset "RNN" begin
|
||||
@testset for R in [RNN, LSTM]
|
||||
@testset for R in [RNN, GRU, LSTM]
|
||||
rnn = R(10, 5)
|
||||
curnn = mapleaves(gpu, rnn)
|
||||
@testset for batch_size in (1, 5)
|
||||
|
|
Loading…
Reference in New Issue