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

don't need mxnet here

Browse Source
This commit is contained in:
Mike J Innes 2017-05-01 12:30:37 +01:00
parent 36f4103d7d
commit 4dc3fa6bd2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/char-rnn.jl
View File

@ -28,7 +28,7 @@ m = mxnet(unroll(model, nunroll))
function sample(model, n, temp = 1) function sample(model, n, temp = 1)
s = [rand(alphabet)] s = [rand(alphabet)]
m = mxnet(unroll1(model)) m = unroll1(model)
for i = 1:n-1 for i = 1:n-1
push!(s, wsample(alphabet, softmax(m(unsqueeze(onehot(s[end], alphabet)))./temp)[1,:])) push!(s, wsample(alphabet, softmax(m(unsqueeze(onehot(s[end], alphabet)))./temp)[1,:]))
end end