better for nested batches

2017-04-19 17:18:40 +01:00 · 2017-04-19 17:18:40 +01:00 · edfb0211e6
commit edfb0211e6
parent 14afe54143
1 changed files with 1 additions and 1 deletions
--- a/src/utils.jl
+++ b/src/utils.jl
@ -5,7 +5,7 @@ const AArray = AbstractArray
 initn(dims...) = randn(dims...)/100

 tobatch(xs::Batch) = rawbatch(xs)
-tobatch(xs) = unsqueeze(xs)
+tobatch(xs) = tobatch(batchone(xs))

 function train!(m, train, test = []; epoch = 1, η = 0.1)
    i = 0