TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix recurrent layer batching

Browse Source
This commit is contained in:
Mike J Innes 2017-03-20 23:11:08 +00:00
parent 6237aa6739
commit eadf96605c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/recurrent.jl
View File

@ -9,7 +9,7 @@ export Recurrent, GatedRecurrent, LSTM
end end
Recurrent(in, out; init = initn) = Recurrent(in, out; init = initn) =
Recurrent(init((in, out)), init((out, out)), init(out), init(out)) Recurrent(init((in, out)), init((out, out)), init(1, out), init(1, out))
@net type GatedRecurrent @net type GatedRecurrent
Wxr; Wyr; br Wxr; Wyr; br