Skip to content

Commit e0f1dab

Browse files
committed
Add option "verbose" for predict api (PaddlePaddle#33405)
* add option verbose for predict api
1 parent 03f4668 commit e0f1dab

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

python/paddle/hapi/model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1831,6 +1831,7 @@ def predict(self,
18311831
batch_size=1,
18321832
num_workers=0,
18331833
stack_outputs=False,
1834+
verbose=1,
18341835
callbacks=None):
18351836
"""
18361837
Compute the output predictions on testing data.
@@ -1851,7 +1852,10 @@ def predict(self,
18511852
be a length N list in shape [[X, Y], [X, Y], ....[X, Y]] if stack_outputs
18521853
is False. stack_outputs as False is used for LoDTensor output situation,
18531854
it is recommended set as True if outputs contains no LoDTensor. Default: False.
1855+
verbose (int): The verbosity mode, should be 0, 1, or 2. 0 = silent,
1856+
1 = progress bar, 2 = one line per batch. Default: 1.
18541857
callbacks(Callback): A Callback instance, default None.
1858+
18551859
Returns:
18561860
list: output of models.
18571861
@@ -1911,7 +1915,7 @@ def __len__(self):
19111915

19121916
self._test_dataloader = test_loader
19131917

1914-
cbks = config_callbacks(callbacks, model=self, verbose=1)
1918+
cbks = config_callbacks(callbacks, model=self, verbose=verbose)
19151919

19161920
test_steps = self._len_data_loader(test_loader)
19171921
logs = {'steps': test_steps}

0 commit comments

Comments
 (0)