Hi,
To answer your question we are not loading our custom trained biobert model from ".from_pretrained. we are using torch.load() to load the  model, as i mentioned i am saving this model in .pt extension. The process we are following below:
we are downloading the pretrained model called “microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract” from model hub, we are applying transfer learning to it with our custom hospital data and then we are saving the model. Now we want to use this saved model to generate the classification labels on sagemaker.
Can you please guide me the way of saving model so that it can be loaded from “.from_pretrained”
Thanks,
Akash