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

tweaks

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 00:34:44 +01:00
parent 1761e43bc4
commit ce0cde0d2b
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/char-rnn.jl
View File

@ -1,7 +1,5 @@
using Flux using Flux
using Juno
getseqs(chars, alphabet) = sequences((onehot(Float32, char, alphabet) for char in chars), 50) getseqs(chars, alphabet) = sequences((onehot(Float32, char, alphabet) for char in chars), 50)
getbatches(chars, alphabet) = batches((getseqs(part, alphabet) for part in chunk(chars, 50))...) getbatches(chars, alphabet) = batches((getseqs(part, alphabet) for part in chunk(chars, 50))...)
@ -17,6 +15,6 @@ model = Chain(
m = tf(unroll(model, 50)) m = tf(unroll(model, 50))
Flux.train!(m, train, η = 0.1/50, epoch = 5) Flux.train!(m, train, η = 0.1/50)
map(c->onecold(c, alphabet), m(train[1][1][1])) map(c->onecold(c, alphabet), m(train[1][1][1]))