Hi! I’ve been trying to train BART on a dialogue data set but got stuck on the following error. I don’t have explicit decoder input ids, so the forward() function calls the _prepare_bart_decoder_inputs() function which in turn calls the shift_tokens_right() function to make the decoder input. The…

Training BART, error when preparing decoder_input_ids. Shape of input_ids?

Valdegg August 6, 2020, 8:19pm 3

Thanks for the answer.

Did you need to change the code in the BART module, or just your own code that uses it?