Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
GPT2_migraphx
Commits
33430a56
Commit
33430a56
authored
Jun 09, 2023
by
liucong
Browse files
修改部分文档
parent
c99ee00b
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
2 additions
and
2 deletions
+2
-2
Doc/Images/GPT_04.png
Doc/Images/GPT_04.png
+0
-0
README.md
README.md
+1
-1
Src/GPT2/gpt2.cpp
Src/GPT2/gpt2.cpp
+1
-1
No files found.
Doc/Images/GPT_04.png
View replaced file @
c99ee00b
View file @
33430a56
35.3 KB
|
W:
|
H:
14.1 KB
|
W:
|
H:
2-up
Swipe
Onion skin
README.md
View file @
33430a56
...
...
@@ -8,7 +8,7 @@ GPT2主要使用Transformer的Decoder模块为特征提取器,并对Transforme
## Python版本推理
本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在model文件夹下。下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python目录中有详细说明。
本次采用GPT-2模型进行诗词生成任务,模型文件下载链接:https://pan.baidu.com/s/1KWeoUuakCZ5dualK69qCcw , 提取码:4pmh ,并将GPT2_shici.onnx模型文件保存在
Resource/
model文件夹下。下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python目录中有详细说明。
### 安装镜像
...
...
Src/GPT2/gpt2.cpp
View file @
33430a56
...
...
@@ -23,7 +23,7 @@ GPT2::~GPT2()
}
ErrorCode
GPT2
::
Initialize
(
InitializationParameterOfNLP
initParamOfNLPGPT2
)
ErrorCode
GPT2
::
Initialize
()
{
// 设置模型路径
std
::
string
modelPath
=
"../Resource/Models/GPT2_shici.onnx"
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment