Skip to content

GitLab

Commit 2cf23df4 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent eba1b174
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
README.md
View file @ 2cf23df4
...@@ -38,11 +38,6 @@ docker run --shm-size 16g --network=host --name=mobilenet_v2 --privileged --devi ...@@ -38,11 +38,6 @@ docker run --shm-size 16g --network=host --name=mobilenet_v2 --privileged --devi
source /opt/dtk/env.sh source /opt/dtk/env.sh
``` ```
### 模型文件
模型文件下载地址：
```
"https://github.com/onnx/models/blob/main/vision/classification/mobilenet/model/mobilenetv2-7.onnx"
```
## 数据集 ## 数据集
python 推理及调优代码使用的图片数据为： python 推理及调优代码使用的图片数据为：
``` ```
...@@ -57,6 +52,12 @@ python 推理及调优代码使用的图片数据为： ...@@ -57,6 +52,12 @@ python 推理及调优代码使用的图片数据为：
"https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip" "https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip"
``` ```
## 推理 ## 推理
### 模型文件
模型文件下载地址：
```
"https://github.com/onnx/models/blob/main/vision/classification/mobilenet/model/mobilenetv2-7.onnx"
```
下载模型文件后执行以下命令进行推理测试及调优测试： 下载模型文件后执行以下命令进行推理测试及调优测试：
``` ```
export MIOPEN_DEBUG_CONV_IMPLICIT_GEMM=0 export MIOPEN_DEBUG_CONV_IMPLICIT_GEMM=0
......
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