add rnn in test - oops

FluxML · Oct 15, 2020 · f6f9925 · f6f9925
1 parent 4d27f4b
commit f6f9925
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/layers/recurrent.jl b/test/layers/recurrent.jl
@@ -1,7 +1,8 @@
 # Ref FluxML/Flux.jl#1209
 @testset "BPTT" begin
   seq = [rand(2) for i = 1:3]
-  for rnn ∈ [RNN,]
+  for r ∈ [RNN,]
+    rnn = r(2,3)
     Flux.reset!(rnn)
     grads_seq = gradient(Flux.params(rnn)) do
       sum(rnn.(seq)[3])