TextClassificationBenchmark
TextClassificationBenchmark copied to clipboard
Add BERT benchmarks
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.
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#
@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.