Support a regression task?

[zhoumengbo]

In analyzing this project, I noticed that the current task description uses "correct/incorrect" to describe the model's output. However, this approach is more suited for classification tasks. I am currently using this project for a task, but my task is a regression task. How should I adjust the model architecture to support a regression task? For example, should the loss function be modified?