diff --git a/python/paddle/hapi/model.py b/python/paddle/hapi/model.py index 40cba4f45d8ef2899ec3c4a66cdf217adc0ed205..e53ab12f841806381b8dfbb9ed6e15c7fe12a64d 100644 --- a/python/paddle/hapi/model.py +++ b/python/paddle/hapi/model.py @@ -1831,6 +1831,7 @@ class Model(object): batch_size=1, num_workers=0, stack_outputs=False, + verbose=1, callbacks=None): """ Compute the output predictions on testing data. @@ -1851,7 +1852,10 @@ class Model(object): be a length N list in shape [[X, Y], [X, Y], ....[X, Y]] if stack_outputs is False. stack_outputs as False is used for LoDTensor output situation, it is recommended set as True if outputs contains no LoDTensor. Default: False. + verbose (int): The verbosity mode, should be 0, 1, or 2. 0 = silent, + 1 = progress bar, 2 = one line per batch. Default: 1. callbacks(Callback): A Callback instance, default None. + Returns: list: output of models. @@ -1911,7 +1915,7 @@ class Model(object): self._test_dataloader = test_loader - cbks = config_callbacks(callbacks, model=self, verbose=1) + cbks = config_callbacks(callbacks, model=self, verbose=verbose) test_steps = self._len_data_loader(test_loader) logs = {'steps': test_steps}