Jelajahi Sumber

disable training for exported inference model

will-jl944 4 tahun lalu
induk
melakukan
031992d6f6
2 mengubah file dengan 8 tambahan dan 0 penghapusan
  1. 4 0
      paddlex/cv/models/detector.py
  2. 4 0
      paddlex/cv/models/segmenter.py

+ 4 - 0
paddlex/cv/models/detector.py

@@ -196,6 +196,10 @@ class BaseDetector(BaseModel):
                 `pretrain_weights` can be set simultaneously. Defaults to None.
 
         """
+        if self.status == 'Infer':
+            logging.error(
+                "Exported inference model does not support training.",
+                exit=True)
         if pretrain_weights is not None and resume_checkpoint is not None:
             logging.error(
                 "pretrain_weights and resume_checkpoint cannot be set simultaneously.",

+ 4 - 0
paddlex/cv/models/segmenter.py

@@ -235,6 +235,10 @@ class BaseSegmenter(BaseModel):
                 `pretrain_weights` can be set simultaneously. Defaults to None.
 
         """
+        if self.status == 'Infer':
+            logging.error(
+                "Exported inference model does not support training.",
+                exit=True)
         if pretrain_weights is not None and resume_checkpoint is not None:
             logging.error(
                 "pretrain_weights and resume_checkpoint cannot be set simultaneously.",