Skip to content

Commit

Permalink
cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
CarloLucibello committed Dec 13, 2024
1 parent 098c064 commit 48e91bc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/layers/recurrent.jl
Original file line number Diff line number Diff line change
Expand Up @@ -307,10 +307,10 @@ end
end

@testset "Recurrence" begin
x = rand(Float32, 2, 3, 4)
for rnn in [RNN(2 => 3), LSTM(2 => 3), GRU(2 => 3)]
cell = rnn.cell
rec = Recurrence(cell)
x = rand(Float32, 2, 3, 4)
@test rec(x) rnn(x)
end
end

0 comments on commit 48e91bc

Please sign in to comment.