From c1a6365a1ef76667cb6ae3cbc08084e2fc6e78f7 Mon Sep 17 00:00:00 2001 From: Pratik Rathore Date: Mon, 25 Nov 2024 11:14:27 -0800 Subject: [PATCH] minor cleanup in burgers example --- examples/pinn_forward/Burgers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/pinn_forward/Burgers.py b/examples/pinn_forward/Burgers.py index 893e70789..7b6883ed1 100644 --- a/examples/pinn_forward/Burgers.py +++ b/examples/pinn_forward/Burgers.py @@ -1,4 +1,5 @@ """Backend supported: tensorflow.compat.v1, tensorflow, pytorch, paddle""" + import deepxde as dde import numpy as np @@ -38,13 +39,12 @@ def pde(x, y): model.train(iterations=15000) model.compile("L-BFGS") losshistory, train_state = model.train() -dde.saveplot(losshistory, train_state, issave=True, isplot=True) # """Backend supported: pytorch""" # # Run NNCG after Adam and L-BFGS # dde.optimizers.set_NNCG_options(rank=50, mu=1e-1) # model.compile("NNCG") # losshistory, train_state = model.train(iterations=1000, display_every=100) -# dde.saveplot(losshistory, train_state, issave=True, isplot=True) +dde.saveplot(losshistory, train_state, issave=True, isplot=True) X, y_true = gen_testdata() y_pred = model.predict(X)