adymaharana / curriculum_learning

9 stars 3 forks source link

KeyError: 'segment_ids' #3

Open ajdesh2000 opened 11 months ago

ajdesh2000 commented 11 months ago

Hi, I'm getting KeyError: 'segment_ids' when running python grid_search_hyperparams_roberta_cosmosqa.py. Looks like some lines were commented out here: https://github.com/adymaharana/curriculum_learning/blob/e5edf7f55ff8128844a370df4837716cccc3a274/roberta/utils_multiple_choice.py#L79

Could you help me with the changes that need to be made to get it to work again?