Merge branch 'minibatch' of https://github.com/YichengDWu/Sophon.jl i…

…nto minibatch
YichengDWu · Oct 21, 2022 · 0bbd77a · 0bbd77a
2 parents 31b4317 + a1dc0c0
commit 0bbd77a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/docs/src/tutorials/burgers.jl b/docs/src/tutorials/burgers.jl
@@ -45,12 +45,9 @@ end
 
 function Sophon.sample(sampler::MyFuncSampler)
     (; n, grf, pts) = sampler
-    xs = ys = [
-        cubic_spline_interpolation(pts, pts .* (1 .- pts) .* sample(grf) .+ 0.1 * randn()),
-    ]
-    for _ in 1:(n - 1)
-        y = cubic_spline_interpolation(pts,
-                                       pts .* (1 .- pts) .* sample(grf) .+ 0.1 * randn())
+    ys = [cubic_spline_interpolation(pts, pts .* (1 .- pts) .* sample(grf) .+ 0.1*randn())]
+    for _ in 1:n-1
+        y = cubic_spline_interpolation(pts, pts .* (1 .- pts) .* sample(grf) .+ 0.1*randn())
         push!(ys, y)
     end
     return ys