• Ethan Perez's avatar
    Don't pass in token_type_ids to BART for GLUE (#8929) · 8dfc8c72
    Ethan Perez authored
    Without this fix, training a `BARTForSequenceClassification` model with `run_pl_glue.py` gives `TypeError: forward() got an unexpected keyword argument 'token_type_ids'`, because BART does not have token_type_ids. I've solved this issue in the same way as it's solved for the "distilbert" model, and I can train BART models on SNLI without errors now.
    8dfc8c72
run_pl_glue.py 7.83 KB