千次曝光出价产生
为了促进中文自然语言处理研究的发展,本项目提供了CPM-LM (2.6B)模型的文本生成代码,可用于文本生成的本地测试,并同时为基础进一步研究零次学习/少次学习等场景。[] [] []
安装
首先安装pytorch等基础依赖,再安装以支持fp16:
pip install -r requirements.txt
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
考虑到apex的安装容易发生问题,我们内置了对应的Docker容器,可以进行快速环境搭建。安装方式如下:
docker pull dmye/cpm:v0
参考运行指令如下:
sudo
2021-08-23 20:41:46
926KB
Python
1