fix logger

Author: HydrogenSulfate

ppsci/solver/solver.py

@@ -219,6 +219,10 @@ def __init__(
             if isinstance(loaded_metric, dict):
                 self.best_metric.update(loaded_metric)
 
+        # init logger without FileHandler if not initialized before
+        if logger._logger is None:
+            logger.init_logger("ppsci", None)
+
         # choosing an appropriate training function for different optimizers
         if isinstance(self.optimizer, optim.LBFGS):
             self.train_epoch_func = ppsci.solver.train.train_LBFGS_epoch_func
@@ -252,8 +256,7 @@ def __init__(
             if version.Version(paddle.__version__) != version.Version("0.0.0")
             else f"develop({paddle.version.commit[:7]})"
         )
-        if logger._logger is not None:
-            logger.info(f"Using paddlepaddle {paddle_version} on device {self.device}")
+        logger.info(f"Using paddlepaddle {paddle_version} on device {self.device}")
 
         self.forward_helper = expression.ExpressionSolver()
 

ppsci/utils/expression.py

@@ -85,11 +85,11 @@ def train_forward(
     def eval_forward(
         self,
         expr_dict: Dict[str, Callable],
-        input_dict: Dict[str, Callable],
+        input_dict: Dict[str, paddle.Tensor],
         model: nn.Layer,
         validator: "validate.Validator",
-        label_dict: Dict[str, Callable],
-        weight_dict: Dict[str, Callable],
+        label_dict: Dict[str, paddle.Tensor],
+        weight_dict: Dict[str, paddle.Tensor],
     ):
         # model forward
         if callable(next(iter(expr_dict.values()))):
@@ -118,7 +118,7 @@ def eval_forward(
     def visu_forward(
         self,
         expr_dict: Dict[str, Callable],
-        input_dict: Dict[str, Callable],
+        input_dict: Dict[str, paddle.Tensor],
         model: nn.Layer,
     ):
         # model forward