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

fix translation

Browse Source
This commit is contained in:
Mike J Innes 2017-08-18 00:31:24 +01:00
parent 574fccd44d
commit b17eb78e0a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/translation.jl
View File

@ -1,7 +1,7 @@
# Based on https://arxiv.org/abs/1409.0473 # Based on https://arxiv.org/abs/1409.0473
using Flux using Flux
using Flux: flip, stateless, broadcastto, ∘ using Flux: Batch, Seq, param, flip, stateless, broadcastto, ∘
Nbatch = 3 # Number of phrases to batch together Nbatch = 3 # Number of phrases to batch together
Nphrase = 5 # The length of (padded) phrases Nphrase = 5 # The length of (padded) phrases
@ -45,6 +45,6 @@ model = @Chain(
model = mxnet(Flux.SeqModel(model, Nphrase)) model = mxnet(Flux.SeqModel(model, Nphrase))
xs = Batch(Seq(rand(Nalpha) for i = 1:Nphrase) for i = 1:Nbatch) xs = Batch([Seq(rand(Nalpha) for i = 1:Nphrase) for i = 1:Nbatch])
model(xs) model(xs)