convert parameters also

This commit is contained in:
Mike J Innes 2017-01-27 16:02:52 +05:30
parent 42fabadd11
commit a71c79e920
2 changed files with 2 additions and 2 deletions

View File

@ -49,7 +49,7 @@ interp(ctx, c::Conv2D, x) =
interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) =
haskey(ctx[:params], p.value) ?
ctx[:params][p.value] :
(ctx[:params][p.value] = Variable(p.value.x))
(ctx[:params][p.value] = Variable(convertel(Float32, p.value.x)))
interp(ctx, p::Constant) = p.value

View File

@ -2,7 +2,7 @@ export AArray
const AArray = AbstractArray
initn(dims...) = randn(Float32, dims...)/10
initn(dims...) = randn(dims...)/10
function train!(m, train, test = []; epoch = 1, batch = 10, η = 0.1)
i = 0