From 74ace2c64422f09913822e18b82040996d30ff91 Mon Sep 17 00:00:00 2001 From: Pratik Rathore Date: Thu, 21 Nov 2024 17:24:54 -0800 Subject: [PATCH] comment out nncg code block --- examples/pinn_forward/Burgers.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/pinn_forward/Burgers.py b/examples/pinn_forward/Burgers.py index ecd4449e2..893e70789 100644 --- a/examples/pinn_forward/Burgers.py +++ b/examples/pinn_forward/Burgers.py @@ -39,12 +39,12 @@ def pde(x, y): 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) +# """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) X, y_true = gen_testdata() y_pred = model.predict(X)