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")