hey @vasudevgupta, is is possible that you’re not providing the correct labels / label names for the loss to be computed?
judging by output_dir, it looks like you’re doing question answering, in which case the trainer looks for the column names ["start_positions", "end_positions"] in the label_names argument.