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 vote96 views1 answer
Get date only from datetimefiel django
Python
0 vote102 views1 answer
Truncate string python
Python
0 vote93 views1 answer
Python3 boto3 put and put_object to s3
Python
0 vote101 views1 answer
Pytorchvideo stack expects each tensor to be equal size, but got [89088] at entry 0 and [88064] at entry 1
Python
2 votes164 views1 answer
Remove negative numbers from list python
Python
2 votes98 views4 answers
Python hash
Python
0 vote106 views1 answer
How to select all but last columns in python
Python
0 vote105 views1 answer
How to modify name of email from divi
Python
0 vote164 views1 answer
Pycache in gitignore
Python
1 vote137 views1 answer
Query with condition django
Python Django