|
@@ -361,7 +361,7 @@ class BaseClassifier(BaseModel):
|
|
|
batch_size_each_card = 1
|
|
|
batch_size = batch_size_each_card * paddlers.env_info['num']
|
|
|
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 " \
|
|
|
"is forcibly set to {}.".format(batch_size))
|
|
|
self.eval_data_loader = self.build_data_loader(
|