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

mxmodel tweaks

Browse Source
This commit is contained in:
Mike J Innes 2016-08-25 17:25:53 +01:00
parent afac5d8bfe
commit 526165c897
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/mxnet/model.jl
View File

@ -1,4 +1,4 @@
type MXModel type MXModel <: Model
model::Any model::Any
params::Dict{Symbol,Any} params::Dict{Symbol,Any}
grads::Dict{Symbol,Any} grads::Dict{Symbol,Any}
@ -59,7 +59,7 @@ end
function (model::MXModel)(input) function (model::MXModel)(input)
tond!(model.exec.arg_dict[:input], input) tond!(model.exec.arg_dict[:input], input)
mx.forward(model.exec) mx.forward(model.exec, is_train = true)
fromnd(model.exec.outputs[1]) fromnd(model.exec.outputs[1])
end end