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

convert parameters also

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/tensorflow/graph.jl
View File

@ -49,7 +49,7 @@ interp(ctx, c::Conv2D, x) =
interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) = interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) =
haskey(ctx[:params], p.value) ? haskey(ctx[:params], p.value) ?
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 interp(ctx, p::Constant) = p.value

2
src/utils.jl
View File

@ -2,7 +2,7 @@ export AArray
const AArray = AbstractArray 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) function train!(m, train, test = []; epoch = 1, batch = 10, η = 0.1)
i = 0 i = 0