├── README.md ├── __init__.py ├── bert_config.py ├── checkpoints ├── cnews │ └── 占位.txt └── cpws │ └── 占位.txt ├── data ├── cnews │ ├── labels.txt │ └── process.py └── cpws │ ├── labels.txt │ └── process.py ├── data_loader.py ├── logs ├── main.log └── preprocess.log ├── main.py ├── main_dataparallel.py ├── main_distributed.py ├── main_mp_distributed.py ├── models.py ├── nvidia.bat ├── requirements.txt ├── test_ddp.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-37.pyc └── utils.cpython-37.pyc └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bert_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/bert_config.py -------------------------------------------------------------------------------- /checkpoints/cnews/占位.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /checkpoints/cpws/占位.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/cnews/labels.txt: -------------------------------------------------------------------------------- 1 | 教育 2 | 娱乐 3 | 家居 4 | 房产 5 | 科技 6 | 时尚 7 | 体育 8 | 财经 9 | 时政 10 | 游戏 11 | -------------------------------------------------------------------------------- /data/cnews/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/data/cnews/process.py -------------------------------------------------------------------------------- /data/cpws/labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/data/cpws/labels.txt -------------------------------------------------------------------------------- /data/cpws/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/data/cpws/process.py -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/data_loader.py -------------------------------------------------------------------------------- /logs/main.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/logs/main.log -------------------------------------------------------------------------------- /logs/preprocess.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/logs/preprocess.log -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/main.py -------------------------------------------------------------------------------- /main_dataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/main_dataparallel.py -------------------------------------------------------------------------------- /main_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/main_distributed.py -------------------------------------------------------------------------------- /main_mp_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/main_mp_distributed.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/models.py -------------------------------------------------------------------------------- /nvidia.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/nvidia.bat -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/test_ddp.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/utils/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taishan1994/pytorch_bert_chinese_text_classification/HEAD/utils/utils.py --------------------------------------------------------------------------------