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 votes150 views2 answers
Check cuda
Python
0 vote164 views2 answers
Convert birth date column to age pandas
Python
0 vote167 views1 answer
Python get the name of the song that is playing
Python
0 vote171 views1 answer
HumanName python
Python
0 vote172 views2 answers
Np.exp in python numpy
Python
0 vote184 views1 answer
Standardscaler
Python
0 vote197 views1 answer
Django create token for user
Python Django
0 vote166 views1 answer
Add new dimension
Python
0 vote165 views1 answer
Multi-line comments Python Example
Python
0 vote189 views2 answers
How to add element to python list
Python