def METHOD_NAME(self):
    return_batch, uttid_list = self.transform(self.data, return_uttid=True)
    batch = self.converter([return_batch], self.device)
    if isinstance(batch, tuple):
        att_ws = self.att_vis_fn(*batch)
    elif isinstance(batch, dict):
        att_ws = self.att_vis_fn(**batch)
    return att_ws, uttid_list