flax-community
/

bert-swahili-news-classification

Text Classification

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

bert-swahili-news-classification / flax_to_torch.py

fgaim's picture

flax-to-torch

5f83f93 almost 3 years ago

raw history blame

No virus

160 Bytes

	from transformers import BertForSequenceClassification

	model = BertForSequenceClassification.from_pretrained("./", from_flax=True)
	model.save_pretrained("./")