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

split basemodel/model

Browse Source
This commit is contained in:
Mike J Innes 2017-02-28 15:53:33 +00:00
parent 209c089e5a
commit 05d988dd74
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/src/examples/char-rnn.md
View File

@ -16,13 +16,15 @@ N = length(alphabet)
Xs, Ys = getbatches(input, alphabet), getbatches(input[2:end], alphabet) Xs, Ys = getbatches(input, alphabet), getbatches(input[2:end], alphabet)
model = Chain( basemodel = Chain(
Input(N), Input(N),
LSTM(N, 256), LSTM(N, 256),
LSTM(256, 256), LSTM(256, 256),
Affine(256, N), Affine(256, N),
softmax) softmax)
model = Chain(basemodel, softmax)
m = tf(unroll(model, nunroll)) m = tf(unroll(model, nunroll))
@time Flux.train!(m, Xs, Ys, η = 0.1, epoch = 1) @time Flux.train!(m, Xs, Ys, η = 0.1, epoch = 1)
@ -36,5 +38,5 @@ function sample(model, n, temp = 1)
return string(s...) return string(s...)
end end
sample(model, 100) sample(basemodel, 100)
``` ```

6
examples/char-rnn.jl
View File

@ -13,13 +13,15 @@ N = length(alphabet)
Xs, Ys = getbatches(input, alphabet), getbatches(input[2:end], alphabet) Xs, Ys = getbatches(input, alphabet), getbatches(input[2:end], alphabet)
model = Chain( basemodel = Chain(
Input(N), Input(N),
LSTM(N, 256), LSTM(N, 256),
LSTM(256, 256), LSTM(256, 256),
Affine(256, N), Affine(256, N),
softmax) softmax)
model = Chain(basemodel, softmax)
m = tf(unroll(model, nunroll)) m = tf(unroll(model, nunroll))
@time Flux.train!(m, Xs, Ys, η = 0.1, epoch = 1) @time Flux.train!(m, Xs, Ys, η = 0.1, epoch = 1)
@ -33,4 +35,4 @@ function sample(model, n, temp = 1)
return string(s...) return string(s...)
end end
sample(model, 100) sample(basemodel, 100)