'Args:
config (PredictConfig): the config of predictor.
dataset (DataFlow): the DataFlow to run on.'

def __init__(self, config, dataset):

    assert isinstance(dataset, DataFlow)
    assert isinstance(config, PredictConfig)
    self.config = config
    self.dataset = dataset
