Skip to content

Commit fa50812

Browse files
Update native.jl batchsize for integer epochs
1 parent ac4d740 commit fa50812

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/native.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ using MLUtils, OptimizationOptimisers
3030

3131
x0 = (-pi):0.001:pi
3232
y0 = sin.(x0)
33-
data = MLUtils.DataLoader((x0, y0), batchsize = 100)
33+
data = MLUtils.DataLoader((x0, y0), batchsize = 126)
3434
function loss(coeffs, data)
3535
ypred = [evalpoly(data[1][i], coeffs) for i in eachindex(data[1])]
3636
return sum(abs2, ypred .- data[2])

0 commit comments

Comments
 (0)