microsoft
/

MiniLM-L12-H384-uncased

Text Classification

Inference Endpoints

Model card Files Files and versions Community

MiniLM-L12-H384-uncased / convert_model.py

system's picture

system HF staff

Update convert_model.py

1654f0f almost 4 years ago

raw history blame contribute delete

No virus

473 Bytes

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

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

	tok = BertTokenizer.from_pretrained("../../MiniLM-L12-H384-uncased/")
	tok.save_pretrained("./")