Save a Bert model with custom forward function and heads on Hugginface

[AkshatSurolia]

Don’t use AutoModel.from_pretrained() for custom models, Use BertClassifier class which you created… so BertClassifier.from_pretrained(‘path’)