BERT

a simple yet complete implementation of the popular BERT model (added with some special treatment to Chinese).

update! I got no idea why this repo suddenly gets some public attention but thanks. I just made an update to make it better.

Experiments show this code could achieve close, if not better, performance to that of Google.

An internal variant of this code produced pre-trained models that are widely used at Tencent.

Advanced Features

• Distributed training
• Lazy file reader

Requirement

• python==3.6
• torch==1.0.0

Quick Guide

• This code is very simple, it should explain itself.

• Train a model from scratch

  • Prepare training corpus and vocab

    • use preprocess.py, see more details there

  • Training

    • sh train.sh

  • For hyper-parameter and all other settings, see the argument parsers in the above two files. We provide a piece of raw text from zhwiki in toy folder.

  • Distributed training settings (in train.sh)

    • --world_size #total number of gpus
    • --gpus #gpus on this machine
    • --MASTER_ADDR #master node IP
    • --MASTER_PORT #master node port
    • --start_rank # range from 0 to world_size-1, the index of the first gpu on this machine
    • --backend # 'nccl' or 'gloo', nccl is generally better but may not work on some machines

• Exemplar use of a trained model

  • See sentence_pair_matching for more details.

  • Preprocessing Guide

    from google_bert import BasicTokenizer
    tokenizer = BasicTokenizer()
    x = "BERT在多个自然语言处理任务中表现优越。"
    char_level_tokens = tokenizer.tokenize(x)
    
    # if processing at word level
    # We assume a word segmenter "word_segmenter" in hand
    word_level_tokens = word_segmenter.segment(x)
    #Note you may need to add speical tokens (e.g., [CLS], [SEP]) by yourself.

Contact

Deng Cai