Mismatched target and input size for BCE using "multi_label_classification"

When defining your model, you did not specify the number of labels (with num_labels=xxx).

4 Likes