paddlex 3.0.0b2__py3-none-any.whl → 3.0.0rc0__py3-none-any.whl

paddlex/modules/open_vocabulary_segmentation/dataset_checker.py

@@ -0,0 +1,29 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .model_list import MODELS
+from ..base import BaseDatasetChecker
+from ...utils.errors import UnsupportedAPIError
+
+
+class OVSegDatasetChecker(BaseDatasetChecker):
+    """Dataset Checker for Open Vocabulary Segmentation Model"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError(
+            "open vocabulary segmentation models do not support data check for now."
+        )

paddlex/modules/open_vocabulary_segmentation/evaluator.py

@@ -0,0 +1,29 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .model_list import MODELS
+from ..base import BaseEvaluator
+from ...utils.errors import UnsupportedAPIError
+
+
+class OVSegEvaluator(BaseEvaluator):
+    """Open Vocabulary Segmentation Model Evaluator"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError(
+            "open vocabulary segmentation models do not support evaluate for now."
+        )

paddlex/modules/open_vocabulary_segmentation/exportor.py

@@ -0,0 +1,29 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .model_list import MODELS
+from ..base import BaseExportor
+from ...utils.errors import UnsupportedAPIError
+
+
+class OVSegExportor(BaseExportor):
+    """Open Vocabulary Segmentation Model Exportor"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError(
+            "open vocabulary segmentation models do not support export for now."
+        )

paddlex/modules/open_vocabulary_segmentation/model_list.py

@@ -0,0 +1,19 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+MODELS = [
+    "SAM-H_point",
+    "SAM-H_box",
+]

paddlex/modules/open_vocabulary_segmentation/trainer.py

@@ -0,0 +1,42 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .model_list import MODELS
+from ..base import BaseTrainer
+from ...utils.errors import UnsupportedAPIError
+
+
+class OVSegTrainer(BaseTrainer):
+    """Open Vocabulary Segmentation Model Trainer"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError(
+            "open vocabulary segmentation models do not support train for now."
+        )
+
+    def update_config(self):
+        """update training config"""
+        pass
+
+    def get_train_kwargs(self) -> dict:
+        """get key-value arguments of model training function
+
+        Returns:
+            dict: the arguments of training function.
+        """
+        train_args = {"device": self.get_device()}
+        return train_args

paddlex/modules/semantic_segmentation/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 
 from ...base import BaseDatasetChecker
 from .dataset_src import check_dataset, convert_dataset, split_dataset, anaylse_dataset
@@ -28,6 +29,20 @@ class SegDatasetChecker(BaseDatasetChecker):
     entities = MODELS
     sample_num = 10
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/images"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 

paddlex/modules/semantic_segmentation/exportor.py

@@ -20,3 +20,12 @@ class SegExportor(BaseExportor):
     """Semantic Segmentation Model Exportor"""
 
     entities = MODELS
+
+    def get_export_kwargs(self):
+        """get key-value arguments of model export function"""
+        kwargs = super().get_export_kwargs()
+
+        input_shape = self.export_config.get("input_shape")
+        if input_shape is not None:
+            kwargs["input_shape"] = input_shape
+        return kwargs

paddlex/modules/semantic_segmentation/model_list.py

@@ -32,4 +32,6 @@ MODELS = [
     "SeaFormer_tiny",
     "SeaFormer_small",
     "SeaFormer_large",
+    "MaskFormer_tiny",
+    "MaskFormer_small",
 ]

paddlex/modules/semantic_segmentation/trainer.py

@@ -68,4 +68,6 @@ class SegTrainer(BaseTrainer):
             train_args["do_eval"] = True
             train_args["save_interval"] = self.train_config.eval_interval
         train_args["dy2st"] = self.train_config.get("dy2st", False)
+        if self.train_config.get("input_shape") is not None:
+            train_args["input_shape"] = self.train_config.input_shape
         return train_args

paddlex/modules/table_recognition/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 from collections import defaultdict, Counter
 from PIL import Image
 import json
@@ -30,6 +31,20 @@ class TableRecDatasetChecker(BaseDatasetChecker):
 
     entities = MODELS
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/train.txt"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 
@@ -64,7 +79,7 @@ class TableRecDatasetChecker(BaseDatasetChecker):
         Returns:
             dict: dataset summary.
         """
-        return check(dataset_dir, self.global_config.output, sample_num=10)
+        return check(dataset_dir, self.output, sample_num=10)
 
     def get_show_type(self) -> str:
         """get the show type of dataset

paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py

@@ -16,6 +16,7 @@
 import os
 import json
 import os.path as osp
+from PIL import Image, ImageOps
 from collections import defaultdict
 from .....utils.errors import DatasetFileNotFoundError, CheckFailedError
 
@@ -58,24 +59,22 @@ def check(dataset_dir, output, dataset_type="PubTabTableRecDataset", sample_num=
                         structure = info["html"]["structure"]["tokens"].copy()
 
                         img_path = osp.join(dataset_dir, file_name)
-                        if len(sample_paths[tag]) < max_recorded_sample_cnts:
-                            sample_paths[tag].append(os.path.relpath(img_path, output))
 
                         if not os.path.exists(img_path):
                             raise DatasetFileNotFoundError(file_path=img_path)
-
-                        boxes_num = len(cells)
-                        tokens_num = sum(
-                            [
-                                structure.count(x)
-                                for x in ["<td>", "<td", "<eb></eb>", "<td></td>"]
-                            ]
-                        )
-                        if boxes_num != tokens_num:
-                            raise CheckFailedError(
-                                f"The number of cells needs to be consistent with the number of tokens "
-                                "but the number of cells is {boxes_num}, and the number of tokens is {tokens_num}."
+                        vis_save_dir = osp.join(output, "demo_img")
+                        if not osp.exists(vis_save_dir):
+                            os.makedirs(vis_save_dir)
+                        if len(sample_paths[tag]) < sample_num:
+                            img = Image.open(img_path)
+                            img = ImageOps.exif_transpose(img)
+                            vis_path = osp.join(vis_save_dir, osp.basename(file_name))
+                            img.save(vis_path)
+                            sample_path = osp.join(
+                                "check_dataset", os.path.relpath(vis_path, output)
                             )
+                            sample_paths[tag].append(sample_path)
+
         meta = {}
 
         meta["train_samples"] = sample_cnts["train"]

paddlex/modules/table_recognition/model_list.py

@@ -16,4 +16,6 @@
 MODELS = [
     "SLANet",
     "SLANet_plus",
+    "SLANeXt_wired",
+    "SLANeXt_wireless",
 ]

paddlex/modules/text_detection/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 from collections import defaultdict, Counter
 
 import json
@@ -30,6 +31,20 @@ class TextDetDatasetChecker(BaseDatasetChecker):
 
     entities = MODELS
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/images"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 
@@ -64,7 +79,7 @@ class TextDetDatasetChecker(BaseDatasetChecker):
         Returns:
             dict: dataset summary.
         """
-        return check(dataset_dir, self.global_config.output, sample_num=10)
+        return check(dataset_dir, self.output, sample_num=10)
 
     def analyse(self, dataset_dir: str) -> dict:
         """deep analyse dataset

paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py

@@ -17,7 +17,7 @@ import os
 import os.path as osp
 from collections import defaultdict
 
-from PIL import Image
+from PIL import Image, ImageOps
 import json
 import numpy as np
 
@@ -66,10 +66,20 @@ def check(dataset_dir, output, sample_num=10):
                     file_name = substr[0]
                     label = substr[1]
                     img_path = osp.join(dataset_dir, file_name)
-                    if len(sample_paths[tag]) < sample_num:
-                        sample_paths[tag].append(os.path.relpath(img_path, output))
                     if not osp.exists(img_path):
                         raise DatasetFileNotFoundError(file_path=img_path)
+                    vis_save_dir = osp.join(output, "demo_img")
+                    if not osp.exists(vis_save_dir):
+                        os.makedirs(vis_save_dir)
+                    if len(sample_paths[tag]) < sample_num:
+                        img = Image.open(img_path)
+                        img = ImageOps.exif_transpose(img)
+                        vis_path = osp.join(vis_save_dir, osp.basename(file_name))
+                        img.save(vis_path)
+                        sample_path = osp.join(
+                            "check_dataset", os.path.relpath(vis_path, output)
+                        )
+                        sample_paths[tag].append(sample_path)
 
                     # check det label
                     label = json.loads(label)

paddlex/modules/text_detection/model_list.py

@@ -17,6 +17,8 @@ MODELS = [
     "PP-OCRv4_server_det",
     "PP-OCRv4_mobile_seal_det",
     "PP-OCRv4_server_seal_det",
+    "PP-OCRv3_mobile_det",
+    "PP-OCRv3_server_det",
 ]
 
 CURVE_MODELS = ["PP-OCRv4_mobile_seal_det", "PP-OCRv4_server_seal_det"]

paddlex/modules/text_recognition/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 from collections import defaultdict, Counter
 
 from PIL import Image
@@ -24,9 +25,6 @@ from ...base import BaseDatasetChecker
 from .dataset_src import check, split_dataset, deep_analyse, convert
 
 from ..model_list import MODELS
-from ...formula_recognition.model_list import MODELS as MODELS_LaTeX
-
-MODELS = MODELS + MODELS_LaTeX
 
 
 class TextRecDatasetChecker(BaseDatasetChecker):
@@ -35,6 +33,20 @@ class TextRecDatasetChecker(BaseDatasetChecker):
     entities = MODELS
     sample_num = 10
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/train.txt"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 
@@ -74,7 +86,7 @@ class TextRecDatasetChecker(BaseDatasetChecker):
         """
         return check(
             dataset_dir,
-            self.global_config.output,
+            self.output,
             sample_num=10,
             dataset_type=self.get_dataset_type(),
         )

paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py

@@ -17,7 +17,7 @@ import os
 import os.path as osp
 from collections import defaultdict
 
-from PIL import Image
+from PIL import Image, ImageOps
 import json
 import numpy as np
 
@@ -79,11 +79,21 @@ def check(
                             )
                         file_name = substr[0]
                         img_path = osp.join(dataset_dir, file_name)
-                        if len(sample_paths[tag]) < max_recorded_sample_cnts:
-                            sample_paths[tag].append(os.path.relpath(img_path, output))
 
                         if not os.path.exists(img_path):
                             raise DatasetFileNotFoundError(file_path=img_path)
+                        vis_save_dir = osp.join(output, "demo_img")
+                        if not osp.exists(vis_save_dir):
+                            os.makedirs(vis_save_dir)
+                        if len(sample_paths[tag]) < sample_num:
+                            img = Image.open(img_path)
+                            img = ImageOps.exif_transpose(img)
+                            vis_path = osp.join(vis_save_dir, osp.basename(file_name))
+                            img.save(vis_path)
+                            sample_path = osp.join(
+                                "check_dataset", os.path.relpath(vis_path, output)
+                            )
+                            sample_paths[tag].append(sample_path)
 
         meta = {}
         meta["train_samples"] = sample_cnts["train"]

paddlex/modules/text_recognition/evaluator.py

@@ -17,9 +17,6 @@ from pathlib import Path
 
 from ..base import BaseEvaluator
 from .model_list import MODELS
-from ..formula_recognition.model_list import MODELS as MODELS_LaTeX
-
-MODELS = MODELS + MODELS_LaTeX
 
 
 class TextRecEvaluator(BaseEvaluator):
@@ -35,6 +32,10 @@ class TextRecEvaluator(BaseEvaluator):
             self.pdx_config.update_dataset(
                 self.global_config.dataset_dir, "LaTeXOCRDataSet"
             )
+        elif "PP-OCRv3" in self.global_config["model"]:
+            self.pdx_config.update_dataset(
+                self.global_config.dataset_dir, "SimpleDataSet"
+            )
         else:
             self.pdx_config.update_dataset(
                 self.global_config.dataset_dir, "MSTextRecDataset"

paddlex/modules/text_recognition/exportor.py

@@ -14,9 +14,6 @@
 
 from ..base import BaseExportor
 from .model_list import MODELS
-from ..formula_recognition.model_list import MODELS as MODELS_LaTeX
-
-MODELS = MODELS + MODELS_LaTeX
 
 
 class TextRecExportor(BaseExportor):

paddlex/modules/text_recognition/model_list.py

@@ -13,8 +13,22 @@
 # limitations under the License.
 
 MODELS = [
+    "PP-OCRv3_mobile_rec",
+    "en_PP-OCRv3_mobile_rec",
+    "korean_PP-OCRv3_mobile_rec",
+    "japan_PP-OCRv3_mobile_rec",
+    "chinese_cht_PP-OCRv3_mobile_rec",
+    "te_PP-OCRv3_mobile_rec",
+    "ka_PP-OCRv3_mobile_rec",
+    "ta_PP-OCRv3_mobile_rec",
+    "latin_PP-OCRv3_mobile_rec",
+    "arabic_PP-OCRv3_mobile_rec",
+    "cyrillic_PP-OCRv3_mobile_rec",
+    "devanagari_PP-OCRv3_mobile_rec",
     "PP-OCRv4_mobile_rec",
     "PP-OCRv4_server_rec",
+    "en_PP-OCRv4_mobile_rec",
+    "PP-OCRv4_server_rec_doc",
     "ch_SVTRv2_rec",
     "ch_RepSVTR_rec",
 ]

paddlex/modules/text_recognition/trainer.py

@@ -20,9 +20,6 @@ from pathlib import Path
 from ..base import BaseTrainer
 from ...utils.config import AttrDict
 from .model_list import MODELS
-from ..formula_recognition.model_list import MODELS as MODELS_LaTeX
-
-MODELS = MODELS + MODELS_LaTeX
 
 
 class TextRecTrainer(BaseTrainer):
@@ -54,6 +51,10 @@ class TextRecTrainer(BaseTrainer):
             self.pdx_config.update_dataset(
                 self.global_config.dataset_dir, "LaTeXOCRDataSet"
             )
+        elif "PP-OCRv3" in self.global_config["model"]:
+            self.pdx_config.update_dataset(
+                self.global_config.dataset_dir, "SimpleDataSet"
+            )
         else:
             self.pdx_config.update_dataset(
                 self.global_config.dataset_dir, "MSTextRecDataset"

paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 from collections import defaultdict, Counter
 
 from PIL import Image
@@ -32,6 +33,20 @@ class TSADDatasetChecker(BaseDatasetChecker):
     entities = MODELS
     sample_num = 10
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/train.csv"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 

paddlex/modules/ts_anomaly_detection/trainer.py

@@ -34,7 +34,21 @@ class TSADTrainer(BaseTrainer):
         os.makedirs(self.global_config.output, exist_ok=True)
         self.update_config()
         self.dump_config()
-        train_result = self.pdx_model.train(**self.get_train_kwargs())
+        train_args = self.get_train_kwargs()
+        export_with_pir = self.global_config.get("export_with_pir", False) or os.getenv(
+            "FLAGS_json_format_model"
+        ) in ["1", "True"]
+        train_args.update(
+            {
+                "uniform_output_enabled": self.train_config.get(
+                    "uniform_output_enabled", True
+                ),
+                "export_with_pir": export_with_pir,
+            }
+        )
+        if self.benchmark_config is not None:
+            train_args.update({"benchmark": self.benchmark_config})
+        train_result = self.pdx_model.train(**train_args)
         assert (
             train_result.returncode == 0
         ), f"Encountered an unexpected error({train_result.returncode}) in \
@@ -80,6 +94,8 @@ training!"
             self.pdx_config.update_learning_rate(self.train_config.learning_rate)
         if self.train_config.epochs_iters is not None:
             self.pdx_config.update_epochs(self.train_config.epochs_iters)
+        if self.train_config.get("dy2st", False):
+            self.pdx_config.update_to_static(self.train_config.dy2st)
         if self.train_config.log_interval is not None:
             self.pdx_config.update_log_interval(self.train_config.log_interval)
         if self.global_config.output is not None:

paddlex/modules/ts_classification/dataset_checker/__init__.py

@@ -15,6 +15,7 @@
 
 import os
 import os.path as osp
+from pathlib import Path
 from collections import defaultdict, Counter
 
 from PIL import Image
@@ -32,6 +33,20 @@ class TSCLSDatasetChecker(BaseDatasetChecker):
     entities = MODELS
     sample_num = 10
 
+    def get_dataset_root(self, dataset_dir: str) -> str:
+        """find the dataset root dir
+
+        Args:
+            dataset_dir (str): the directory that contain dataset.
+
+        Returns:
+            str: the root directory of dataset.
+        """
+        anno_dirs = list(Path(dataset_dir).glob("**/train.csv"))
+        assert len(anno_dirs) == 1
+        dataset_dir = anno_dirs[0].parent.as_posix()
+        return dataset_dir
+
     def convert_dataset(self, src_dataset_dir: str) -> str:
         """convert the dataset from other type to specified type
 

paddlex/modules/ts_classification/trainer.py

@@ -34,7 +34,21 @@ class TSCLSTrainer(BaseTrainer):
         os.makedirs(self.global_config.output, exist_ok=True)
         self.update_config()
         self.dump_config()
-        train_result = self.pdx_model.train(**self.get_train_kwargs())
+        train_args = self.get_train_kwargs()
+        export_with_pir = self.global_config.get("export_with_pir", False) or os.getenv(
+            "FLAGS_json_format_model"
+        ) in ["1", "True"]
+        train_args.update(
+            {
+                "uniform_output_enabled": self.train_config.get(
+                    "uniform_output_enabled", True
+                ),
+                "export_with_pir": export_with_pir,
+            }
+        )
+        if self.benchmark_config is not None:
+            train_args.update({"benchmark": self.benchmark_config})
+        train_result = self.pdx_model.train(**train_args)
         assert (
             train_result.returncode == 0
         ), f"Encountered an unexpected error({train_result.returncode}) in \
@@ -77,6 +91,8 @@ training!"
             self.pdx_config.update_epochs(self.train_config.epochs_iters)
         if self.train_config.log_interval is not None:
             self.pdx_config.update_log_interval(self.train_config.log_interval)
+        if self.train_config.get("dy2st", False):
+            self.pdx_config.update_to_static(self.train_config.dy2st)
         if self.global_config.output is not None:
             self.pdx_config.update_save_dir(self.global_config.output)