How to use SaProt as an encoder just like esm

[KyrieYsPKU]

I want to use this model as an encoder for my own model training.But I wonder the meaning of the prompt "You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference" and how to solve this problem.

[LTEnjoy]

Hi,

The prompt means that you have initialized a classifier behind the SaProt backbone. If you want to use it to make predictions, you have to fine-tune the classifier on your own data so that the model will predict accurately. This is not an error but a warning that tell you not to use it directly unless you have trained the model.