I want to do Multiclass-Multilabel ( MLMC)
classification problem using Conv-BERT
model.
Steps that I have taken is:
I downloaded the Conv-Bert model from this link: YituTech/conv-bert-base · Hugging Face
<< YituTech/conv-bert-base>>
from pytorch_pretrained_bert import BertTokenizer, BertForSequenceClassification, BertAdam
tokenizer = BertTokenizer.from_pretrained("path_to_Conv-Bert_model", do_lower_case = True)
model = BertForSequenceClassification.from_pretrained("path_to_Conv-Bert_model", num_labels = 240)
model.cuda()
I want to understand can we call any classification module from Hugging face and pass any pre-trained models to it like Roberta, Conv-BERT.. so on
. ? << As in above example>>