Yizhou Chen пре 3 година
родитељ
комит
39253066a2
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      paddlers/tasks/classifier.py

+ 1 - 1
paddlers/tasks/classifier.py

@@ -361,7 +361,7 @@ class BaseClassifier(BaseModel):
             batch_size_each_card = 1
             batch_size_each_card = 1
             batch_size = batch_size_each_card * paddlers.env_info['num']
             batch_size = batch_size_each_card * paddlers.env_info['num']
             logging.warning(
             logging.warning(
-                "Segmenter only supports batch_size=1 for each gpu/cpu card " \
+                "Classifier only supports batch_size=1 for each gpu/cpu card " \
                 "during evaluation, so batch_size " \
                 "during evaluation, so batch_size " \
                 "is forcibly set to {}.".format(batch_size))
                 "is forcibly set to {}.".format(batch_size))
         self.eval_data_loader = self.build_data_loader(
         self.eval_data_loader = self.build_data_loader(