Mar 27, 2023
Hello Xngli. You're right. MSELoss() is used by default when num_labels=1. We will add a remark in the article, and still keep the RegressionTrainer for reference in case readers would like to apply another loss function.
Written by La Javaness R&D
We help organizations to succeed in the new paradigm of “AI@scale”, by using machine intelligence responsibly and efficiently : www.lajavaness.com