Google bert MRPC task; Custom predictions
Task
Sentence (and sentence-pair) classification
This example code fine-tunes BERT-Base on the Microsoft Research Paraphrase Corpus (MRPC) corpus, which only contains 3,600 examples and can fine-tune in a few minutes on most GPUs.
Environment
docker: tensorflow gpu jupyter
docker hub repo: andrewmoses/rtx2060_tensorflow
GPU: RTX 2060 6GB