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

4 votes264 views1 answer
Python format only 1 decimal place
Python
0 vote320 views1 answer
Python tkinter cursor types
Python
0 vote293 views3 answers
Duck typing in python
Python
0 vote316 views1 answer
Flask request post with json
Python Flask
0 vote306 views1 answer
Creating empty set and append python
Python
15 votes277 views6 answers
Dataframe to list
Python
0 vote287 views1 answer
Is file a keywoard in python
Python
15 votes260 views3 answers
How to import numpy in python
Python
4 votes269 views2 answers
Pandas rolling
Python
0 vote233 views1 answer
Cairo.context transform vertical text python
Python