LeeJuly30 / BERTCpp

implement bert in pure c++

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BERTCpp

A lightly BERT inference project using intel MKL and Protobuf in c++ (working in progress)

Dependency

Protobuf

bertcpp using protobuf to convert pytorch pretrained model in pb file and load it in c++

MKL

bertcpp using MKL to implement blas operator

utf8proc

bertcpp using utf8proc to process string

Build

mkdir build & cd build
cmake ..
make -j4

To run unitest

./bert_test

Thanks

tokenizer part comes from cuBERT

About

implement bert in pure c++

License:MIT License


Languages

Language:C++ 77.4%Language:Python 18.2%Language:CMake 4.4%