From cde74337c810a6d612a476c47c53690344f966d6 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sun, 12 Mar 2017 18:33:29 +0000 Subject: [PATCH] should've moved this --- 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 e74619f9..a6d614bc 100644 --- a/src/backend/mxnet/model.jl +++ b/src/backend/mxnet/model.jl @@ -31,7 +31,7 @@ end ndparams(d::Dict{Symbol,MXArray}) = Dict(k => v.data for (k, v) in d) -type Exec <: Flux.Model +type Exec graph::Graph exec::mx.Executor args::Dict{Symbol,MXArray} @@ -84,7 +84,7 @@ end # TODO: if `last` changes, update params appropriately -type Model +type Model <: Flux.Model model::Any graph::Graph execs::Dict{Tuple,Exec}