Perhaps your features (`output` in this case) have excessive nesting (inputs type `list` where type `int` is expected)

🤗Transformers
4

I think there is also a way to implement and specify the DataCollator that is suitable for each data set, but in some simple cases, it seems that you can deal with it by renaming the column names of the dataset.

1 Like