"vscode:/vscode.git/clone" did not exist on "93d3d8feda296c1c9a78a1d20b37f83cd4299e5b"
Commit 794e44ea authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 25004606
...@@ -20,7 +20,7 @@ ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯 ...@@ -20,7 +20,7 @@ ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯
``` ```
创建并启动容器 创建并启动容器
``` ```
docker run --shm-size 16g --network=host --name=ResNet50_v2 --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /ResNet50_v2_tvm:/home/ResNet50_v2_tvm -it <Your Image ID> /bin/bash docker run --shm-size 16g --network=host --name=ResNet50_v2 --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it <Your Image ID> /bin/bash
# 激活dtk # 激活dtk
source /opt/dtk/env.sh source /opt/dtk/env.sh
...@@ -41,19 +41,10 @@ source /opt/dtk/env.sh ...@@ -41,19 +41,10 @@ source /opt/dtk/env.sh
coco128数据集的目录结构如下: coco128数据集的目录结构如下:
├── images ├── images
│ ├── train2017 │ ├── train2017
│ ├── val2017
│ ├── test2017
├── labels ├── labels
│ ├── train2017 │ ├── train2017
│ ├── val2017
├── annotations
│ ├── instances_val2017.json
├── LICENSE ├── LICENSE
├── README.txt ├── README.txt
├── test-dev2017.txt
├── train2017.txt
├── val2017.txt
``` ```
## 推理 ## 推理
...@@ -66,7 +57,7 @@ source /opt/dtk/env.sh ...@@ -66,7 +57,7 @@ source /opt/dtk/env.sh
git clone http://developer.hpccube.com/codes/modelzoo/tvm_tune_resnet50-v2.git git clone http://developer.hpccube.com/codes/modelzoo/tvm_tune_resnet50-v2.git
cd tvm_tune_resnet50-v2 cd tvm_tune_resnet50-v2
export HIP_VISIBLE_DEVICES=1 #此处为指定哪张加速卡进行推理的ID号 export HIP_VISIBLE_DEVICES=1 #此处为指定哪张加速卡进行推理的ID号
python tune_resnet50-v2.py python tune_resnet50-v2.py #根据实际修改模型路径
``` ```
...@@ -76,7 +67,7 @@ source /opt/dtk/env.sh ...@@ -76,7 +67,7 @@ source /opt/dtk/env.sh
``` ```
mkdir -p lib mkdir -p lib
export HIP_VISIBLE_DEVICES=1 #此处为指定哪张加速卡进行推理的ID号 export HIP_VISIBLE_DEVICES=1 #此处为指定哪张加速卡进行推理的ID号
python prepare_test_libs.py python prepare_test_libs.py #根据实际修改模型路径
sh run_example.sh sh run_example.sh
``` ```
## result ## result
......
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