update code

HydrogenSulfate · HydrogenSulfate · commit 1fefb1062690 · 2024-04-25T11:14:39.000Z
diff --git a/docs/zh/examples/amgnet.md b/docs/zh/examples/amgnet.md
@@ -103,17 +103,17 @@ unzip data.zip
 
 === "airfoil"
 
-    ``` py linenums="61"
+    ``` py linenums="55"
     --8<--
-    examples/amgnet/amgnet_airfoil.py:61:62
+    examples/amgnet/amgnet_airfoil.py:55:56
     --8<--
     ```
 
 === "cylinder"
 
-    ``` py linenums="61"
+    ``` py linenums="55"
     --8<--
-    examples/amgnet/amgnet_cylinder.py:61:62
+    examples/amgnet/amgnet_cylinder.py:55:56
     --8<--
     ```
 
diff --git a/examples/NLS-MB/NLS-MB_optical_rogue_wave.py b/examples/NLS-MB/NLS-MB_optical_rogue_wave.py
@@ -240,7 +240,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -266,7 +265,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.lbfgs.eval_during_train,
         eval_freq=cfg.TRAIN.lbfgs.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -344,7 +342,6 @@ def evaluate(cfg: DictConfig):
         output_dir=cfg.output_dir,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/NLS-MB/NLS-MB_optical_soliton.py b/examples/NLS-MB/NLS-MB_optical_soliton.py
@@ -217,7 +217,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -243,7 +242,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.lbfgs.eval_during_train,
         eval_freq=cfg.TRAIN.lbfgs.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -321,7 +319,6 @@ def evaluate(cfg: DictConfig):
         output_dir=cfg.output_dir,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/allen_cahn/allen_cahn_causal.py b/examples/allen_cahn/allen_cahn_causal.py
@@ -167,7 +167,6 @@ def gen_label_batch(input_batch):
         log_freq=cfg.log_freq,
         eval_during_train=True,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
         validator=validator,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
diff --git a/examples/allen_cahn/allen_cahn_plain.py b/examples/allen_cahn/allen_cahn_plain.py
@@ -165,7 +165,6 @@ def gen_label_batch(input_batch):
         log_freq=cfg.log_freq,
         eval_during_train=True,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
         validator=validator,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
diff --git a/examples/amgnet/amgnet_airfoil.py b/examples/amgnet/amgnet_airfoil.py
@@ -167,7 +167,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
diff --git a/examples/amgnet/amgnet_cylinder.py b/examples/amgnet/amgnet_cylinder.py
@@ -167,7 +167,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
diff --git a/examples/aneurysm/aneurysm.py b/examples/aneurysm/aneurysm.py
@@ -231,9 +231,7 @@ def inlet_w_ref_func(_in):
         log_freq=cfg.log_freq,
         eval_during_train=True,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
@@ -321,7 +319,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
diff --git a/examples/aneurysm/aneurysm_flow.py b/examples/aneurysm/aneurysm_flow.py
@@ -278,7 +278,6 @@ def output_transform_p(self, in_, out):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )
diff --git a/examples/biharmonic2d/biharmonic2d.py b/examples/biharmonic2d/biharmonic2d.py
@@ -231,9 +231,7 @@ def train(cfg: DictConfig):
         cfg.TRAIN.iters_per_epoch,
         save_freq=cfg.TRAIN.save_freq,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
     )
@@ -252,9 +250,7 @@ def train(cfg: DictConfig):
         1,
         save_freq=cfg.TRAIN.save_freq,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
     )
diff --git a/examples/bracket/bracket.py b/examples/bracket/bracket.py
@@ -329,9 +329,7 @@ def train(cfg: DictConfig):
         log_freq=cfg.log_freq,
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
@@ -497,7 +495,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
diff --git a/examples/bubble/bubble.py b/examples/bubble/bubble.py
@@ -186,7 +186,6 @@ def transform_out(in_, out):
         cfg.TRAIN.iters_per_epoch,
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -341,7 +340,6 @@ def transform_out(in_, out):
     solver = ppsci.solver.Solver(
         model_list,
         output_dir=cfg.output_dir,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/cfdgcn/cfdgcn.py b/examples/cfdgcn/cfdgcn.py
@@ -208,7 +208,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
diff --git a/examples/control_arm/forward_analysis.py b/examples/control_arm/forward_analysis.py
@@ -184,9 +184,7 @@ def train(cfg: DictConfig):
         lr_scheduler,
         cfg.TRAIN.epochs,
         cfg.TRAIN.iters_per_epoch,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         save_freq=cfg.TRAIN.save_freq,
         log_freq=cfg.log_freq,
         eval_freq=cfg.TRAIN.eval_freq,
@@ -256,8 +254,6 @@ def evaluate(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model_list,
         output_dir=cfg.output_dir,
-        seed=cfg.seed,
-        geom=geom,
         log_freq=cfg.log_freq,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
         visualizer=visualizer,
diff --git a/examples/control_arm/inverse_parameter.py b/examples/control_arm/inverse_parameter.py
@@ -134,9 +134,7 @@ def train(cfg: DictConfig):
         lr_scheduler,
         cfg.TRAIN.epochs,
         cfg.TRAIN.iters_per_epoch,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         save_freq=cfg.TRAIN.save_freq,
         log_freq=cfg.log_freq,
         eval_freq=cfg.TRAIN.eval_freq,
@@ -228,7 +226,6 @@ def evaluate(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model,
         output_dir=cfg.output_dir,
-        seed=cfg.seed,
         log_freq=cfg.log_freq,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
         validator=validator,
diff --git a/examples/cylinder/2d_unsteady/cylinder2d_unsteady_Re100.py b/examples/cylinder/2d_unsteady/cylinder2d_unsteady_Re100.py
@@ -207,7 +207,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
@@ -284,7 +283,6 @@ def evaluate(cfg: DictConfig):
     # initialize solver
     solver = ppsci.solver.Solver(
         model,
-        geom=geom,
         output_dir=cfg.output_dir,
         validator=validator,
         visualizer=visualizer,
diff --git a/examples/darcy/darcy2d.py b/examples/darcy/darcy2d.py
@@ -151,7 +151,6 @@ def poisson_ref_compute_func(_in):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
     )
@@ -180,7 +179,6 @@ def poisson_ref_compute_func(_in):
         eval_during_train=cfg.TRAIN.lbfgs.eval_during_train,
         eval_freq=cfg.TRAIN.lbfgs.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
     )
@@ -274,7 +272,6 @@ def poisson_ref_compute_func(_in):
         model,
         output_dir=cfg.output_dir,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
diff --git a/examples/deepcfd/deepcfd.py b/examples/deepcfd/deepcfd.py
@@ -311,7 +311,6 @@ def metric_expr(
         epochs=cfg.TRAIN.epochs,
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         validator=validator,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
@@ -413,7 +412,6 @@ def metric_expr(
     solver = ppsci.solver.Solver(
         model,
         output_dir=cfg.output_dir,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
diff --git a/examples/euler_beam/euler_beam.py b/examples/euler_beam/euler_beam.py
@@ -114,9 +114,7 @@ def u_solution_func(out):
         iters_per_epoch=cfg.TRAIN.iters_per_epoch,
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
@@ -184,9 +182,7 @@ def u_solution_func(out):
         None,
         cfg.output_dir,
         None,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         validator=validator,
         visualizer=visualizer,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
diff --git a/examples/fourcastnet/train_pretrain.py b/examples/fourcastnet/train_pretrain.py
@@ -158,7 +158,6 @@ def train(cfg: DictConfig):
         cfg.TRAIN.epochs,
         ITERS_PER_EPOCH,
         eval_during_train=True,
-        seed=cfg.seed,
         validator=validator,
         compute_metric_by_batch=cfg.EVAL.compute_metric_by_batch,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
@@ -231,7 +230,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
         compute_metric_by_batch=cfg.EVAL.compute_metric_by_batch,
diff --git a/examples/fpde/fractional_poisson_2d.py b/examples/fpde/fractional_poisson_2d.py
@@ -173,7 +173,6 @@ def input_data_fpde_transform(
         eval_during_train=True,
         eval_freq=EVAL_FREQ,
         equation=equation,
-        geom=geom,
         validator=validator,
         eval_with_no_grad=True,
     )
diff --git a/examples/fsi/viv.py b/examples/fsi/viv.py
@@ -111,7 +111,6 @@ def train(cfg: DictConfig):
         log_freq=cfg.log_freq,
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
-        seed=cfg.seed,
         equation=equation,
         validator=validator,
         visualizer=visualizer,
diff --git a/examples/gpinn/poisson_1d.py b/examples/gpinn/poisson_1d.py
@@ -138,7 +138,6 @@ def u_solution(in_):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
@@ -255,7 +254,6 @@ def u_solution(in_):
     solver = ppsci.solver.Solver(
         model,
         output_dir=cfg.output_dir,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/heat_exchanger/heat_exchanger.py b/examples/heat_exchanger/heat_exchanger.py
@@ -338,7 +338,6 @@ def train(cfg: DictConfig):
         eval_during_train=cfg.TRAIN.eval_during_train,
         eval_freq=cfg.TRAIN.eval_freq,
         equation=equation,
-        geom=geom,
         validator=validator,
     )
     # train model
@@ -542,7 +541,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         equation=equation,
-        geom=geom,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/heat_pinn/heat_pinn.py b/examples/heat_pinn/heat_pinn.py
@@ -117,9 +117,7 @@ def train(cfg: DictConfig):
         iters_per_epoch=cfg.TRAIN.iters_per_epoch,
         save_freq=cfg.TRAIN.save_freq,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         equation=equation,
-        geom=geom,
         pretrained_model_path=cfg.TRAIN.pretrained_model_path,
         checkpoint_path=cfg.TRAIN.checkpoint_path,
     )
@@ -215,7 +213,6 @@ def evaluate(cfg: DictConfig):
         model,
         output_dir=cfg.output_dir,
         log_freq=cfg.log_freq,
-        seed=cfg.seed,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
     # begin eval
diff --git a/examples/hpinns/holography.py b/examples/hpinns/holography.py
@@ -391,7 +391,6 @@ def evaluate(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model_list,
         output_dir=cfg.output_dir,
-        seed=cfg.seed,
         validator=validator,
         pretrained_model_path=cfg.EVAL.pretrained_model_path,
     )
diff --git a/examples/ide/volterra_ide.py b/examples/ide/volterra_ide.py
diff --git a/examples/laplace/laplace2d.py b/examples/laplace/laplace2d.py
diff --git a/examples/ldc/ldc2d_steady_Re10.py b/examples/ldc/ldc2d_steady_Re10.py
diff --git a/examples/ldc/ldc2d_unsteady_Re10.py b/examples/ldc/ldc2d_unsteady_Re10.py
diff --git a/examples/nsfnet/VP_NSFNet1.py b/examples/nsfnet/VP_NSFNet1.py
diff --git a/examples/nsfnet/VP_NSFNet2.py b/examples/nsfnet/VP_NSFNet2.py
diff --git a/examples/nsfnet/VP_NSFNet3.py b/examples/nsfnet/VP_NSFNet3.py
diff --git a/examples/nsfnet/VP_NSFNet4.py b/examples/nsfnet/VP_NSFNet4.py
diff --git a/examples/operator_learning/deeponet.py b/examples/operator_learning/deeponet.py
diff --git a/examples/phylstm/phylstm2.py b/examples/phylstm/phylstm2.py
diff --git a/examples/phylstm/phylstm3.py b/examples/phylstm/phylstm3.py
diff --git a/examples/shock_wave/shock_wave.py b/examples/shock_wave/shock_wave.py
diff --git a/ppsci/solver/solver.py b/ppsci/solver/solver.py

Original file line number	Diff line number	Diff line change
`@@ -278,7 +278,6 @@ def output_transform_p(self, in_, out):`
`278`	`278`	`model,`
`279`	`279`	`output_dir=cfg.output_dir,`
`280`	`280`	`log_freq=cfg.log_freq,`
`281`		`- seed=cfg.seed,`
`282`	`281`	`pretrained_model_path=cfg.EVAL.pretrained_model_path,`
`283`	`282`	`eval_with_no_grad=cfg.EVAL.eval_with_no_grad,`
`284`	`283`	`)`