Key Error ‘loss’ - finetuning [ arabert , mbert ]
Hello all!I am trying to fine-tune mbert and arabert models on translation task as explainedhere, however, I am getting this errorKey Error ‘loss’the input to the model :DatasetDict({
    train: Dataset({
        features: ['attention_mask', 'input_ids', 'labels', 'src', 'token_type_ids', 'trg'],
        num_rows: 40
    })
    validation: Dataset({
        features: ['attention_mask', 'input_ids', 'labels', 'src', 'token_type_ids', 'trg'],
        num_rows: 11
    })
})Note: when I runtrainer.train(), it returns a message likeThe following columns in the training set  don't have a corresponding argument inBertModel.forwardand have been ignored: trg, labels, src.any hints?
2021-12-23T16:23:03Z
You can use theTraineronBertModelas it has no objective, it’s the body of the model with no particular head. You should pick a model with a head suitable for the task at hand.