fix translation

FluxML · Aug 17, 2017 · b17eb78 · b17eb78
1 parent 574fccd
commit b17eb78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/translation.jl b/examples/translation.jl
@@ -1,7 +1,7 @@
 # Based on https://arxiv.org/abs/1409.0473
 
 using Flux
-using Flux: flip, stateless, broadcastto, ∘
+using Flux: Batch, Seq, param, flip, stateless, broadcastto, ∘
 
 Nbatch  =  3 # Number of phrases to batch together
 Nphrase =  5 # The length of (padded) phrases
@@ -45,6 +45,6 @@ model = @Chain(
 
 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)