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

2 votes224 views3 answers
Local variable
Python
0 vote228 views1 answer
Bold some letters of string in python
Python
0 vote299 views1 answer
Python , cv2 change font type
Python
5 votes226 views1 answer
UnicodeDecodeError: 'charmap' codec can't decode byte 0x9e in position 3359: character maps to <undefined>
Python
6 votes218 views3 answers
Extract frames from video python
Python
0 vote233 views1 answer
Python iterate over line of gzip file
Python
0 vote272 views1 answer
Print chr character in python f string
Python
1 vote183 views3 answers
Python is float function
Python
0 vote241 views1 answer
Json python no whitespace
Python
0 vote229 views1 answer
Pydub reproduce audiosegment
Python