This commit is contained in:
Yingda Chen
2024-11-21 19:59:19 +08:00
parent 3acf9d4e52
commit be04c33912

View File

@@ -197,7 +197,9 @@ class Pipeline(ABC):
# simple showcase, need to support iterator type for both tensorflow and pytorch
# input_dict = self._handle_input(input)
print('__call__')
print(input)
print(kwargs)
# sanitize the parameters
batch_size = kwargs.pop('batch_size', None)
preprocess_params, forward_params, postprocess_params = self._sanitize_parameters(