comment out nncg code block

lululxvi · Nov 22, 2024 · 74ace2c · 74ace2c
1 parent f2d5c31
commit 74ace2c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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)