space_test = sampling.Rectangle([0, 0], [upper, upper], 0.0)
cond_i = sampling.dirichlet([space], uv_func, N0)
#dt = 0.3
#dt = 0.5
#dt = 0.3
#dt = 1.0
dt = 2.0
#dt = 0.6
#dt = 1e-2
x_step = 4e-3
q = 500
layers = [2, 50, 50, 50, 50, 50, 50, (q + 1) * 2]
#layers = [2, 200, 200, 200, 200, (q+1)*2]
net = NeuralNet.Linear(layers)
main1 = main1_2d
model = pde.PINN_Burgers
param = [Re]
#
#elif True:
elif False:
Re = 0.001
func = func_burgers2(Re)
uv_func = translate_func_2d(func)
N0 = 200
Nb = 100
upper = 0.5
