namisan / mt-dnn

Multi-Task Deep Neural Networks for Natural Language Understanding
MIT License
2.24k stars 412 forks source link

Unable to get the complete model MT_DNN #235

Open avyavkumar opened 2 years ago

avyavkumar commented 2 years ago

For branch v0.1, I want to get the complete pre-trained model without task-specific layer from the checkpoint file mt_dnn_base_uncased.pt, however, I am not able to convert it to MTDNNModel. Is there a script that can help me get the multi-task trained MT-DNN model from the checkpoint file? Thanks!