├── README.md ├── config ├── fastbert_cls.json └── fastbert_cls_ernie.json ├── data_utils ├── dataset_preparing.py └── tokenization.py ├── img ├── arch.png └── compare.png ├── infer.py ├── model_define ├── model_bert.py ├── model_fastbert.py └── optimization.py ├── predict.py ├── pretrained_model ├── bert-chinese │ └── bert-pytorch-google │ │ ├── bert_config.json │ │ └── vocab.txt └── ernie │ └── ERNIE_stable-1.0.1-pytorch │ ├── config.json │ └── vocab.txt ├── run_scripts ├── script_eval.sh ├── script_infer.sh ├── script_train_stage0.sh └── script_train_stage1.sh ├── sample └── ChnSentiCorp │ ├── dev.tsv │ ├── test.tsv │ ├── test_slim.tsv │ └── train.tsv ├── saved_model └── README.md ├── train.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/README.md -------------------------------------------------------------------------------- /config/fastbert_cls.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/config/fastbert_cls.json -------------------------------------------------------------------------------- /config/fastbert_cls_ernie.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/config/fastbert_cls_ernie.json -------------------------------------------------------------------------------- /data_utils/dataset_preparing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/data_utils/dataset_preparing.py -------------------------------------------------------------------------------- /data_utils/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/data_utils/tokenization.py -------------------------------------------------------------------------------- /img/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/img/arch.png -------------------------------------------------------------------------------- /img/compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/img/compare.png -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/infer.py -------------------------------------------------------------------------------- /model_define/model_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/model_define/model_bert.py -------------------------------------------------------------------------------- /model_define/model_fastbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/model_define/model_fastbert.py -------------------------------------------------------------------------------- /model_define/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/model_define/optimization.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/predict.py -------------------------------------------------------------------------------- /pretrained_model/bert-chinese/bert-pytorch-google/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/pretrained_model/bert-chinese/bert-pytorch-google/bert_config.json -------------------------------------------------------------------------------- /pretrained_model/bert-chinese/bert-pytorch-google/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/pretrained_model/bert-chinese/bert-pytorch-google/vocab.txt -------------------------------------------------------------------------------- /pretrained_model/ernie/ERNIE_stable-1.0.1-pytorch/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/pretrained_model/ernie/ERNIE_stable-1.0.1-pytorch/config.json -------------------------------------------------------------------------------- /pretrained_model/ernie/ERNIE_stable-1.0.1-pytorch/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/pretrained_model/ernie/ERNIE_stable-1.0.1-pytorch/vocab.txt -------------------------------------------------------------------------------- /run_scripts/script_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/run_scripts/script_eval.sh -------------------------------------------------------------------------------- /run_scripts/script_infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/run_scripts/script_infer.sh -------------------------------------------------------------------------------- /run_scripts/script_train_stage0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/run_scripts/script_train_stage0.sh -------------------------------------------------------------------------------- /run_scripts/script_train_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/run_scripts/script_train_stage1.sh -------------------------------------------------------------------------------- /sample/ChnSentiCorp/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/sample/ChnSentiCorp/dev.tsv -------------------------------------------------------------------------------- /sample/ChnSentiCorp/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/sample/ChnSentiCorp/test.tsv -------------------------------------------------------------------------------- /sample/ChnSentiCorp/test_slim.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/sample/ChnSentiCorp/test_slim.tsv -------------------------------------------------------------------------------- /sample/ChnSentiCorp/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/sample/ChnSentiCorp/train.tsv -------------------------------------------------------------------------------- /saved_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/saved_model/README.md -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BitVoyage/FastBERT/HEAD/utils.py --------------------------------------------------------------------------------