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

3 votes229 views2 answers
How to make python code faster
Python
1 vote187 views1 answer
Torch concat two 1d tensor
Python
0 vote309 views1 answer
Python glob subdirectories
Python
0 vote227 views1 answer
Time complexity for random in python
Python
0 vote211 views1 answer
Python print char n times
Python
2 votes195 views2 answers
Densenet python keras
Python
5 votes241 views4 answers
Dropna for specific column pandas
Python
4 votes197 views2 answers
Median of a list python
Python
1 vote196 views1 answer
Python turtle shooting game
Python
-1 vote190 views1 answer
Data.head on terminal
Python