Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Take an argmax on the last dimension of both the start_logits and end_logits to get the predicted start_idx and end_idx.