Transformers bert

>>> from transformers import BertTokenizer, BertModel
>>> import torch

>>> tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
>>> model = BertModel.from_pretrained('bert-base-uncased')

>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs)

>>> last_hidden_states = outputs.last_hidden_state

Similar codes

0 vote119 views1 answer
Np.all
Python
0 vote116 views1 answer
Split by backslash python
Python
36 votes125 views3 answers
Python image library
Python
0 vote92 views1 answer
Tkinter set text
Python
0 vote113 views1 answer
Python syntax error jedi
Python
2 votes142 views5 answers
List append python 3
Python
0 vote103 views1 answer
Python edit global variable in function
Python
1 vote112 views1 answer
Multirow np.rand.randint
Python
2 votes104 views3 answers
Descending python dataframe df
Python
0 vote114 views1 answer
Mongoengine ObjectIdField
Python