update char-rnn

This commit is contained in:
Mike J Innes 2017-04-28 17:14:35 +01:00
parent 360d128b1a
commit f50688e75c

View File

@ -24,7 +24,7 @@ model = Chain(
m = mxnet(unroll(model, nunroll))
@time Flux.train!(m, train, η = 1, epoch = 1)
@time Flux.train!(m, train, η = 0.1, loss = Flux.logloss)
function sample(model, n, temp = 1)
s = [rand(alphabet)]