bhadresh-savani
/

electra-base-emotion

Text Classification

Inference Endpoints

Model card Files Files and versions Community

electra-base-emotion / convert_pytorch_to_tensorflow.py

bhadresh-savani's picture

bhadresh-savani

added tf and flax model support

d2b8df9 about 2 years ago

raw history blame contribute delete

No virus

166 Bytes

	from transformers import TFElectraForSequenceClassification
	model = TFElectraForSequenceClassification.from_pretrained("./", from_pt=True)
	model.save_pretrained("./")