From b17eb78e0a7131de18501b621c594a9c4d459372 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 18 Aug 2017 00:31:24 +0100 Subject: [PATCH] fix translation --- examples/translation.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/translation.jl b/examples/translation.jl index 89129ad2..abbd84b9 100644 --- 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)