Skip to content

CPU version of BERT QA fine-tuned on SQuAD v1.1 (with sklearn wrapper)

Compare
Choose a tag to compare
@andrelmfarias andrelmfarias released this 31 May 12:02

Release with a version of BERT model trained on SQuAD 1.1 runnable on CPU.

Version available in two formats:

Pytorch model: bert_qa_vCPU.bin and config.json

Sklearn wrapper: bert_qa_vCPU-sklearn.joblib