UniMol2 to fine-tune and inference in the regression(property predict)

[chengfengke]

When I use UniMol2 to fine-tune and inference (regression task), it output is weird, only focus on a small scale,first picture is real data, second is predicted picture

What should I do to solve the problem? btw, Thanks a lot for such wonderful work!