Merge pull request #572 from SciML/ap/minibatch-path

Fix #570
SciML · Jun 17, 2021 · fa82cdb · fa82cdb
2 parents 2f1bfbc + ad2aeb5
commit fa82cdb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/src/examples/minibatch.md b/docs/src/examples/minibatch.md
@@ -46,7 +46,7 @@ ode_data = Array(solve(true_prob, Tsit5(), saveat=t))
 prob = ODEProblem{false}(dudt_, u0, tspan, θ)
 
 k = 10
-train_loader = Flux.Data.DataLoader(ode_data, t, batchsize = k)
+train_loader = Flux.Data.DataLoader((ode_data, t), batchsize = k)
 
 for (x, y) in train_loader
     @show x
@@ -154,7 +154,7 @@ ode_data = Array(solve(true_prob, Tsit5(), saveat=t))
 prob = ODEProblem{false}(dudt_, u0, tspan, θ)
 
 k = 10
-train_loader = Flux.Data.DataLoader(ode_data, t, batchsize = k)
+train_loader = Flux.Data.DataLoader((ode_data, t), batchsize = k)
 for (x, y) in train_loader
     @show x
     @show y