microsoft
/

Multilingual-MiniLM-L12-H384

Text Classification

Model card Files Files and versions

Multilingual-MiniLM-L12-H384 / convert_model.py

system's picture

system HF Staff

Update convert_model.py

1a8eef3 about 6 years ago

History Blame Contribute Delete

538 Bytes

	#!/usr/bin/env python3
	import logging
	from transformers import BertModel, BertTokenizer, TFBertModel, XLMRobertaTokenizer

	logging.basicConfig(level=logging.INFO, filename="log.txt")
	model = BertModel.from_pretrained("../../Multilingual-MiniLM-L12-H384/")
	tf_model = TFBertModel.from_pretrained("../../Multilingual-MiniLM-L12-H384/", from_pt=True)
	model.save_pretrained("./")
	tf_model.save_pretrained("./")

	tok = XLMRobertaTokenizer.from_pretrained("../../Multilingual-MiniLM-L12-H384/sentencepiece.bpe.model")
	tok.save_pretrained("./")