How to use RMSE loss for regression

[adenhaus]

Hi, I’m fine-tuning a transformer (ForSequenceClassification variant) as a regression task, but I want to use RMSE instead of MSE as the loss function, because the paper I am trying to replicate does so. How can I do this?

Thanks