Parcourir la source

disable training for exported inference model

will-jl944 il y a 4 ans
Parent
commit
031992d6f6
2 fichiers modifiés avec 8 ajouts et 0 suppressions
  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.",