code
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||
1. 深度学习框架:
1) pytorch
2. 运行环境:
1) python 3.6.5
2) pytorch 1.1.0
3) CUDNN 7
4) CUDA 9.0
3. 依赖包:
pytorch_pretrained_bert 0.6.2
ujson 1.35
tqdm 4.23.4
PyNLPIR 0.5.2
pkuseg 0.0.22
records 0.5.2
4. 依赖包安装命令:
pip install -r requirements.txt
pynlpir update
chmod +x train.sh
chmod +x predict.sh
5. 文件说明:
训练文件:train.py 执行脚本(./train.sh)
注:如果--bS 32 参数(batch size)会耗尽计算资源,请减小该参数,如将bS设置为16
推理文件:generate_result.py 执行脚本(./predict.sh)