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

export recurrent

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 11:41:52 +00:00
parent fabff9b0d2
commit 81d9743836
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/char-rnn.jl
View File

@ -10,7 +10,7 @@ train = zip(getbatches(input, alphabet), getbatches(input[2:end], alphabet))
model = Chain( model = Chain(
Input(length(alphabet)), Input(length(alphabet)),
Flux.Recurrent(length(alphabet), 128, length(alphabet)), Recurrent(length(alphabet), 128, length(alphabet)),
softmax) softmax)
m = tf(unroll(model, 50)) m = tf(unroll(model, 50))

2
src/layers/recurrent.jl
View File

@ -1,3 +1,5 @@
export Recurrent
@net type Recurrent @net type Recurrent
Wxh; Whh; Why Wxh; Whh; Why
bh; by bh; by