update training args

Author: foldfelis

example/burgers.jl

@@ -15,11 +15,11 @@ loss(𝐱, 𝐲) = sum(abs2, 𝐲 .- m(𝐱)) / size(𝐱)[end]
 
 𝐱, 𝐲 = get_burgers_data(n=2048)
 
-n_train = 2000
+n_train = 1000
 𝐱_train, 𝐲_train = 𝐱[:, :, 1:n_train], 𝐲[:, 1:n_train]
 loader_train = Flux.DataLoader((𝐱_train, 𝐲_train), batchsize=20, shuffle=true)
 
-n_test = 40
+n_test = 100
 𝐱_test, 𝐲_test = 𝐱[:, :, end-n_test+1:end], 𝐲[:, end-n_test+1:end]
 loader_test = Flux.DataLoader((𝐱_test, 𝐲_test), batchsize=20, shuffle=false)