search

FreedomIntelligence / TextClassificationBenchmark

A Benchmark of Text Classification in PyTorch
MIT License
600 stars 138 forks source link

Add BERT benchmarks #21

Open loretoparisi opened 5 years ago

loretoparisi commented 5 years ago

Thanks for this repo. It could be very interesting to add the latest Google's BERT model that claims to be state-of-the-art in recent NLP tasks among them text classification. They have a classifier implementation to adapt, here some details.

wabyking commented 5 years ago

Actually, we are doting this, Please refer to https://github.com/wabyking/TextClassificationBenchmark/blob/master/models/BERTFast.py It depends on a related project named #pytorch_pretrained_bert#

loretoparisi commented 5 years ago

@wabyking thank you! I have just seen that 👍
By the way I was actually referring to BERT for Tensorflow rather than Pytorch as in the original implementation.