restore some config

Author: HydrogenSulfate

examples/NLS-MB/NLS-MB_optical_rogue_wave.py

@@ -261,7 +261,6 @@ def train(cfg: DictConfig):
         eval_freq=cfg.TRAIN.lbfgs.eval_freq,
         equation=equation,
         validator=validator,
-        cfg=cfg,
     )
     # train model
     solver.train()

examples/NLS-MB/NLS-MB_optical_soliton.py

@@ -238,7 +238,6 @@ def train(cfg: DictConfig):
         eval_freq=cfg.TRAIN.lbfgs.eval_freq,
         equation=equation,
         validator=validator,
-        cfg=cfg,
     )
     # train model
     solver.train()

examples/aneurysm/aneurysm_flow.py

@@ -196,10 +196,15 @@ def output_transform_p(self, in_, out):
     solver = ppsci.solver.Solver(
         model,
         constraint,
-        optimizer=optimizer,
+        cfg.output_dir,
+        optimizer,
+        log_freq=cfg.log_freq,
+        epochs=cfg.TRAIN.epochs,
         iters_per_epoch=int(x.shape[0] / cfg.TRAIN.batch_size),
+        save_freq=cfg.save_freq,
         equation=equation,
-        cfg=cfg,
+        pretrained_model_path=cfg.TRAIN.pretrained_model_path,
+        checkpoint_path=cfg.TRAIN.checkpoint_path,
     )
     solver.train()
 

examples/fourcastnet/train_finetune.py

@@ -162,10 +162,15 @@ def train(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model,
         constraint,
-        optimizer=optimizer,
+        cfg.output_dir,
+        optimizer,
+        epochs=cfg.TRAIN.epochs,
         iters_per_epoch=ITERS_PER_EPOCH,
+        eval_during_train=True,
         validator=validator,
-        cfg=cfg,
+        pretrained_model_path=cfg.TRAIN.pretrained_model_path,
+        compute_metric_by_batch=cfg.EVAL.compute_metric_by_batch,
+        eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )
     # train model
     solver.train()

examples/fourcastnet/train_precip.py

@@ -159,10 +159,15 @@ def train(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model,
         constraint,
-        iters_per_epoch=ITERS_PER_EPOCH,
-        optimizer=optimizer,
+        cfg.output_dir,
+        optimizer,
+        lr_scheduler,
+        cfg.TRAIN.epochs,
+        ITERS_PER_EPOCH,
+        eval_during_train=True,
         validator=validator,
-        cfg=cfg,
+        compute_metric_by_batch=cfg.EVAL.compute_metric_by_batch,
+        eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )
     # train model
     solver.train()

examples/fourcastnet/train_pretrain.py

@@ -152,10 +152,16 @@ def train(cfg: DictConfig):
     solver = ppsci.solver.Solver(
         model,
         constraint,
-        iters_per_epoch=ITERS_PER_EPOCH,
-        optimizer=optimizer,
+        cfg.output_dir,
+        optimizer,
+        lr_scheduler,
+        cfg.TRAIN.epochs,
+        ITERS_PER_EPOCH,
+        eval_during_train=True,
+        seed=cfg.seed,
         validator=validator,
-        cfg=cfg,
+        compute_metric_by_batch=cfg.EVAL.compute_metric_by_batch,
+        eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )
     # train model
     solver.train()