change predictor

Author: wufei2

examples/cylinder/2d_unsteady/transformer_physx/train_transformer.py

@@ -292,9 +292,22 @@ def export(cfg: DictConfig):
 
 
 def inference(cfg: DictConfig):
-    from deploy.python_infer import pinn_predictor
-
-    predictor = pinn_predictor.PINNPredictor(cfg)
+    from deploy.python_infer import base
+
+    predictor = base.Predictor(
+        cfg.INFER.pdmodel_path,
+        cfg.INFER.pdiparams_path,
+        device=cfg.INFER.device,
+        engine=cfg.INFER.engine,
+        precision=cfg.INFER.precision,
+        onnx_path=cfg.INFER.onnx_path,
+        ir_optim=cfg.INFER.ir_optim,
+        min_subgraph_size=cfg.INFER.min_subgraph_size,
+        gpu_mem=cfg.INFER.gpu_mem,
+        gpu_id=cfg.INFER.gpu_id,
+        max_batch_size=cfg.INFER.max_batch_size,
+        num_cpu_threads=cfg.INFER.num_cpu_threads,
+    )
 
     dataset_cfg = {
         "name": "CylinderDataset",