Commit b6e7ad7a authored by yangql's avatar yangql
Browse files

修改格式

parent 074dd4bd
...@@ -25,9 +25,17 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0 ...@@ -25,9 +25,17 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0
``` ```
docker run --shm-size 16g --network=host --name=gpt2_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/gpt2_ort:/home/gpt2_ort -it <Your Image ID> /bin/bash docker run --shm-size 16g --network=host --name=gpt2_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/gpt2_ort:/home/gpt2_ort -it <Your Image ID> /bin/bash
# 安装依赖
pip uninstall urllib3
pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 激活dtk # 激活dtk
source /opt/dtk/env.sh source /opt/dtk/env.sh
``` ```
## 数据集
采用交互式界面,通过输入开头诗词,GPT2模型可以推理出后续的诗句。
## 推理 ## 推理
### Python版本推理 ### Python版本推理
本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在Resource/文件夹下。下面介绍如何运行python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。 本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在Resource/文件夹下。下面介绍如何运行python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。
...@@ -52,16 +60,6 @@ python gpt2.py ...@@ -52,16 +60,6 @@ python gpt2.py
``` ```
如下所示,采用交互式界面,通过输入开头诗词,GPT2模型可以生成后续的诗句。 如下所示,采用交互式界面,通过输入开头诗词,GPT2模型可以生成后续的诗句。
```
user:江上归帆天际开
chatbot:江上归帆天际开,江头别棹日边回。风尘满地音书绝,鸿雁不来春又来。
user:我亦孤山冷淡郎
chatbot:我亦孤山冷淡郎,爱梅不作一般香。水边篱落坡仙笑,羔酒空浇入甲黄。
user:七言绝句
chatbot:七言绝句古无有,五字长城今在前。我欲从君乞妙语,笔端三昧要亲传。
user:春风吹絮满江南
chatbot:春风吹絮满江南,一片离情酒半酣。记得小桥和雪看,梅花无数簇晴岚。
```
### C++版本推理 ### C++版本推理
本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在Resource/文件夹下。下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。 本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在Resource/文件夹下。下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。
...@@ -94,8 +92,19 @@ cd build/ ...@@ -94,8 +92,19 @@ cd build/
./GPT2 ./GPT2
``` ```
如下所示,采用交互式界面,通过输入开头诗词,GPT2模型可以推理出后续的诗句。 ## result
### python版本
```
user:江上归帆天际开
chatbot:江上归帆天际开,江头别棹日边回。风尘满地音书绝,鸿雁不来春又来。
user:我亦孤山冷淡郎
chatbot:我亦孤山冷淡郎,爱梅不作一般香。水边篱落坡仙笑,羔酒空浇入甲黄。
user:七言绝句
chatbot:七言绝句古无有,五字长城今在前。我欲从君乞妙语,笔端三昧要亲传。
user:春风吹絮满江南
chatbot:春风吹絮满江南,一片离情酒半酣。记得小桥和雪看,梅花无数簇晴岚。
```
### C++版本
``` ```
question:江上归帆天际开 question:江上归帆天际开
chatbot:江上归帆天际开,江头别棹日边回。风尘满地音书绝,鸿雁不来春又来。 chatbot:江上归帆天际开,江头别棹日边回。风尘满地音书绝,鸿雁不来春又来。
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment