From 1cc8100456c01ae02c714980bc7a8d2c9b08bf72 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 9 Jun 2017 01:35:07 +0100 Subject: [PATCH] ctx methods for seq models --- src/backend/mxnet/model.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/mxnet/model.jl b/src/backend/mxnet/model.jl index 19fe53a1..6a09d44f 100644 --- a/src/backend/mxnet/model.jl +++ b/src/backend/mxnet/model.jl @@ -124,8 +124,8 @@ Flux.update!(m::Model, η) = (update!(m.last, η); m) using Flux: Stateful, SeqModel -mxnet(m::Stateful) = Stateful(mxnet(m.model), m.states, m.istate, m.ostate) -mxnet(m::SeqModel) = SeqModel(mxnet(m.model), m.steps) +mxnet(m::Stateful, a...) = Stateful(mxnet(m.model, a...), m.states, m.istate, m.ostate) +mxnet(m::SeqModel, a...) = SeqModel(mxnet(m.model, a...), m.steps) # MX FeedForward interface