Skip to content

[Future] Enable Users to Use Models Off-the-shelf #26

@abheesht17

Description

@abheesht17

Often, users don't want to train the model; they just want to load the pretrained weights (which we can provide - we can perhaps upload them on Google Drive) and use the model for inference. So, we need to write a generic .from_pretrained() function. The usage of this can be something like this:

from src.models.caml import CAML
trained_model = CAML.from_pretrained("<weight-file-path>.pt")

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions