JavaBERT-small / convert_to_TF.py
anjandash's picture
added model files
af38e1d
raw
history blame contribute delete
270 Bytes
from transformers import TFAutoModelForSequenceClassification
checkpoint_path = "/Users/anjandash/Desktop/HF_MODELS/JavaBERT-small"
tf_model = TFAutoModelForSequenceClassification.from_pretrained(checkpoint_path, from_pt=True)
tf_model.save_pretrained(checkpoint_path)