update char-rnn
This commit is contained in:
parent
360d128b1a
commit
f50688e75c
@ -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)]
|
||||
|
Loading…
Reference in New Issue
Block a user