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

better for nested batches

Browse Source
This commit is contained in:
Mike J Innes 2017-04-19 17:18:40 +01:00
parent 14afe54143
commit edfb0211e6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/utils.jl
View File

@ -5,7 +5,7 @@ const AArray = AbstractArray
initn(dims...) = randn(dims...)/100 initn(dims...) = randn(dims...)/100
tobatch(xs::Batch) = rawbatch(xs) tobatch(xs::Batch) = rawbatch(xs)
tobatch(xs) = unsqueeze(xs) tobatch(xs) = tobatch(batchone(xs))
function train!(m, train, test = []; epoch = 1, η = 0.1) function train!(m, train, test = []; epoch = 1, η = 0.1)
i = 0 i = 0