def unpack_batch(batch, use_cuda=False):
    """ Unpack a batch from the data loader. """
    input_ids = batch[0]
    input_mask = batch[1]
    segment_ids = batch[2]
    boundary_ids = batch[3]
    pos_ids = batch[4]
    rel_ids = batch[5]
    knowledge_feature = batch[6]
    bio_ids = batch[1]
    # knowledge_adjoin_matrix = batch[7]
    # know_segment_ids = batch[6]
    # know_input_ids = batch[7]
    # know_input_mask = batch[8]
    # knowledge_feature = (batch[6], batch[7], batch[8])

    return input_ids, input_mask, segment_ids, boundary_ids, pos_ids, rel_ids, knowledge_feature,bio_ids