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
template_xxx
Commits
ad42e33a
Commit
ad42e33a
authored
Mar 16, 2023
by
liuhy
Browse files
Initial commit
parents
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
83 additions
and
0 deletions
+83
-0
README.md
README.md
+83
-0
No files found.
README.md
0 → 100644
View file @
ad42e33a
# 模型名称(此处需修改,用英文全称与简写)
## 模型介绍
此处填写模型介绍。
## 模型结构
此处简要介绍模型结构。
## 数据集
此处介绍使用的数据集
如添加了自己写的数据集处理脚本,在此处说明脚本的使用方法:
python xxx.py \
--args0 xxx \
--args1 xxx \
...
## 训练及推理
### 环境配置
提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取的训练以及推理的docker镜像:
*
训练镜像:
*
推理镜像:
### 训练与Fine-tunning
训练命令:
python train.py \
--args0 xxx \
--args1 xxx \
...
Fine-tunning命令:
python train.py \
--args0 xxx \
--args1 xxx \
...
### 预训练模型
model文件夹提供的预训练模型介绍,例如:
Project
├── model
│ ├── xxx.pth #pytorch预训练模型
│ ├── xxx.onnx #对应的onnx模型
└── └── xxx.mxr #对应的migraphx离线推理模型
### 测试
测试命令:
python test.py \
--args0 xxx \
--args1 xxx
...
### 推理
推理引擎版本:
*
ONNXRuntime(DCU版本) >= x.xx.x
*
Migraphx(DCU版本) >= x.x.x
#### ORT
基于ORT的推理命令:
python ORT_infer.py \
--args0 xxx \
--args1 xxx \
...
#### Migraphx
基于Migraphx的推理命令:
python Migraphx_infer.py \
--args0 xxx \
--args1 xxx \
...
## 性能和准确率数据
测试数据:
[
test data
](
链接
)
,使用的加速卡:xxx。
根据模型情况填写表格:
| xxx | xxx | xxx | xxx | xxx |
| :------: | :------: | :------: | :------: |:------: |
| xxx | xxx | xxx | xxx | xxx |
| xxx | xx | xxx | xxx | xxx |
## 历史版本
*
仓库的https链接
## 参考
*
[
Reference1
](
链接
)
*
[
Reference2
](
链接
)
*
......
\ No newline at end of file
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