convert some model to pytorch
convert tensorflow checkpoint to pytorch.
The origin checkpoint from https://github.com/ymcui/Chinese-MobileBERT
With Huggingface-Transformers, the models above could be easily accessed and loaded through the following codes.
tokenizer = MobileBertTokenizer.from_pretrained("MODEL_NAME")
model = MobileBertModel.from_pretrained("MODEL_NAME")
**Notice: Please use MobileBertTokenizer and MobileBertModel for loading these model. **
The actual model and its MODEL_NAME are listed below.
| Original Model | MODEL_NAME |
|---|---|
| Chinese-MobileBERT-base-f2 | cycloneboy/chinese_mobilebert_base_f2 |
| Chinese-MobileBERT-base-f4 | cycloneboy/chinese_mobilebert_base_f4 |
| Chinese-MobileBERT-large-f2 | cycloneboy/chinese_mobilebert_large_f2 |
| Chinese-MobileBERT-large-f4 | cycloneboy/chinese_mobilebert_large_f4 |
convert some model to pytorch
convert tensorflow checkpoint to pytorch.
The origin checkpoint from https://github.com/ymcui/Chinese-MobileBERT
With Huggingface-Transformers, the models above could be easily accessed and loaded through the following codes.
tokenizer = MobileBertTokenizer.from_pretrained("MODEL_NAME")
model = MobileBertModel.from_pretrained("MODEL_NAME")
**Notice: Please use MobileBertTokenizer and MobileBertModel for loading these model. **
The actual model and its MODEL_NAME are listed below.
| Original Model | MODEL_NAME |
|---|---|
| Chinese-MobileBERT-base-f2 | cycloneboy/chinese_mobilebert_base_f2 |
| Chinese-MobileBERT-base-f4 | cycloneboy/chinese_mobilebert_base_f4 |
| Chinese-MobileBERT-large-f2 | cycloneboy/chinese_mobilebert_large_f2 |
| Chinese-MobileBERT-large-f4 | cycloneboy/chinese_mobilebert_large_f4 |