Update test_pinn.py

2023-08-02 10:04:02 +02:00
parent e84def3bf9
commit 8c16e27ae4
1 changed files with 0 additions and 1 deletions
--- a/tests/test_solvers/test_pinn.py
+++ b/tests/test_solvers/test_pinn.py
@@ -104,7 +104,6 @@ def test_train_restore():
    pinn = PINN(problem = poisson_problem, model=model, extra_features=None, loss=LpLoss())
    trainer = Trainer(solver=pinn, max_epochs=5, accelerator='cpu', default_root_dir=tmpdir)
    trainer.train()
-    print('ggg')
    ntrainer = Trainer(solver=pinn, max_epochs=15, accelerator='cpu')
    t = ntrainer.train(
        ckpt_path=f'{tmpdir}/lightning_logs/version_0/checkpoints/epoch=4-step=5.ckpt')