diff --git a/examples/char-rnn.jl b/examples/char-rnn.jl index 78c04033..f0536319 100644 --- a/examples/char-rnn.jl +++ b/examples/char-rnn.jl @@ -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)]