Adding Custom label names for BART training through Trainer Function

Hi all I am trying to add label_names in Trainer for BART model. Now since I have used custom label names my BART is throwing error in the model(**inputs) since its not able to handle the custom label names that I passed through Trainer. Is there way to handle multi labels in BART

1 Like

Hello, have you solved your problem? I tried out-of-label classification results when using Bart to do multi-classification problems.