Problem when saving the model !

hiyouga / Dual-Contrastive-Learning

Code for our paper "Dual Contrastive Learning: Text Classification via Label-Aware Data Augmentation"

https://arxiv.org/abs/2201.08702

MIT License

156 stars 28 forks source link

Problem when saving the model ! #15

Open SoDehghan opened 1 year ago

SoDehghan commented 1 year ago

Hi, thank you for your exciting work. When I want to save model get this error: Transformer object has no "save_pretrained"

How to save the model after we train it with your code? In fact, I want to save the model and upload it in Huggingface so that I can load and use it later.