Skip to content

Help with Converting Protein Sequence to Input Data for ESM_S Model #2

@deardeer7

Description

@deardeer7

Hello,

I'm new to the whole torchdrug pipeline thing. I was checking out the README and saw this part about "Load Trained Model Weight." After loading the weights, how do I go about turning a protein sequence into the input data, feed it into the esm_s model, and then grab the output embeddings? I reviewed the code, and there's this ESM class with a forward function that looks like forward(self, graph, input,..). Can you help me figure out how to put together the graph and input from a protein sequence? Any tips or pointers would be greatly appreciated.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions