bert-for-inference
bert-for-inference copied to clipboard
BramVanroy
A small repo showing how to easily use BERT (or other transformers) for inference
Results
1
bert-for-inference issues
Sort by
recently updated
recently updated
newest added
I am running your code and found there is one issue : `sentence_embedding = torch.mean(hidden_states[-1], dim=1).squeeze()` should be `sentence_embedding = torch.mean(hidden_states[-1], dim=0).squeeze()` Please check!
Metadata
94
Stars
21
Forks
Watchers
Owner
Metadata
A small repo showing how to easily use BERT (or other transformers) for inference