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
ResNet50-v2_tvm
Commits
794e44ea
Commit
794e44ea
authored
Dec 01, 2023
by
dcuai
Browse files
Update README.md
parent
25004606
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
12 deletions
+3
-12
README.md
README.md
+3
-12
No files found.
README.md
View file @
794e44ea
...
...
@@ -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
...
...
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