from transformers import BertForSequenceClassification def get_full_finetune_model(): model = BertForSequenceClassification.from_pretrained('bert-base-uncased') return model