search

codertimo / BERT-pytorch

Google AI 2018 BERT pytorch implementation
Apache License 2.0
6.11k stars 1.29k forks source link

Single Sentence Input support #31

Closed nateraw closed 5 years ago

nateraw commented 5 years ago

In the paper, they note that they optionally use single sentence input for some classification tasks. I'll try to take a look at doing it myself, as it looks like it is not currently supported.

codertimo commented 5 years ago

@nateraw Well currently, BERT-pytorch only support 2sentences input. Cause this implementation is only for training BERT, not for transfer learning :)

codertimo commented 5 years ago

But you can give a single sentence, if you change little bit of code 👍