forked from benzakenelad/BitFit
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest.py
More file actions
15 lines (11 loc) · 690 Bytes
/
test.py
File metadata and controls
15 lines (11 loc) · 690 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from transformers import AutoModelForSequenceClassification, AutoConfig, DebertaForSequenceClassification, DistilBertForSequenceClassification
# model_name = 'distilbert-base-uncased'
model_name = 'microsoft/deberta-base'
config = AutoConfig.from_pretrained(model_name, num_labels=2, return_dict=True)
model = AutoModelForSequenceClassification.from_pretrained(model_name, config=config)
# model = DistilBertForSequenceClassification.from_pretrained(model_name, return_dict=True).distilbert
# model = DebertaForSequenceClassification.from_pretrained(model_name, return_dict=True).deberta
# print(model)
# print(model.deberta)
for name, param in model.named_parameters():
print(name)