fix

Author: HydrogenSulfate

ppsci/solver/solver.py

@@ -1037,5 +1037,5 @@ def _parse_params_from_cfg(self, cfg: DictConfig):
             self.pretrained_model_path = cfg.TRAIN.pretrained_model_path
         elif cfg.mode == "eval":
             self.pretrained_model_path = cfg.EVAL.pretrained_model_path
-        elif cfg.mode == "infer":
+        elif cfg.mode in ["export", "infer"]:
             self.pretrained_model_path = cfg.INFER.pretrained_model_path