Hi.
I used the pre-trained weights of the DistilBERT model in order to train a text classification model with the SST-2 dataset. Now, when I am trying to export the trained model to SavedModel format, it fails. Here’s 1 the full error trace.
Here’s the notebook 2 for the training code. I use the following lines of code in order to export the trained model to SavedModel -
!mkdir DistilBERT
model.save("DistilBERT")