def METHOD_NAME(self, args, data_loader):
    start_time = time.time()
    for i, data in enumerate(data_loader):
        if i < args.perf_warmup_steps:  # skip warmup steps.
            continue
        output = self.predict_batch([data["input_ids"], data["token_type_ids"]])
        paddle.to_tensor(output)
    used_time = time.time() - start_time
    return used_time