项目地址:https://github.com/Morizeyao/GPT2-Chinese
所有命令
创建虚拟环境
conda create -n gpt2 python=3.10
激活虚拟环境
conda activate gpt2
移动到程序目录
cd F:\素材\深度学习\gpt2\GPT2
安装默认环境
pip install transformers==2.1.1 --trusted-host pypi.douban.com
pip install torch --trusted-host pypi.douban.com
pip install numpy --trusted-host pypi.douban.com
pip install tqdm --trusted-host pypi.douban.com
pip install sklearn --trusted-host pypi.douban.com
pip install keras --trusted-host pypi.douban.com
pip install tb-nightly --trusted-host pypi.douban.com
pip install future --trusted-host pypi.douban.com
pip install thulac --trusted-host pypi.douban.com
运行py文件
python ./generate.py --length=50 --nsamples=4 --prefix=东风软软百花香 --fast_pattern --save_samples --save_samples_path=F:\素材\深度学习\gpt2\GPT2\output\歌词
python ./generate.py --length=250 --nsamples=4 --prefix=xxx --fast_pattern --save_samples --save_samples_path=F:\素材\深度学习\gpt2\GPT2\output\