Commit 794e44ea authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 25004606
......@@ -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
source /opt/dtk/env.sh
......@@ -41,19 +41,10 @@ source /opt/dtk/env.sh
coco128数据集的目录结构如下:
├── images
│ ├── train2017
│ ├── val2017
│ ├── test2017
├── labels
│ ├── train2017
│ ├── val2017
├── annotations
│ ├── instances_val2017.json
├── LICENSE
├── README.txt
├── test-dev2017.txt
├── train2017.txt
├── val2017.txt
```
## 推理
......@@ -66,7 +57,7 @@ source /opt/dtk/env.sh
git clone http://developer.hpccube.com/codes/modelzoo/tvm_tune_resnet50-v2.git
cd tvm_tune_resnet50-v2
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
```
mkdir -p lib
export HIP_VISIBLE_DEVICES=1 #此处为指定哪张加速卡进行推理的ID号
python prepare_test_libs.py
python prepare_test_libs.py #根据实际修改模型路径
sh run_example.sh
```
## 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