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_onnxruntime
Commits
17fa8aed
Commit
17fa8aed
authored
Oct 17, 2023
by
yangql
Browse files
新增方法二
parent
9a9a0d62
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
1 deletion
+13
-1
README.md
README.md
+10
-1
docker/Dockerfile
docker/Dockerfile
+3
-0
No files found.
README.md
View file @
17fa8aed
...
@@ -15,7 +15,7 @@ ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯
...
@@ -15,7 +15,7 @@ ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯


## 环境配置
## 环境配置
### Docker
### Docker
(方法一)
拉取镜像:
拉取镜像:
```
python
```
python
docker
pull
image
.
sourcefind
.
cn
:
5000
/
dcu
/
admin
/
base
/
custom
:
ort1
.
14.0
_migraphx3
.
0.0
-
dtk22
.
10.1
docker
pull
image
.
sourcefind
.
cn
:
5000
/
dcu
/
admin
/
base
/
custom
:
ort1
.
14.0
_migraphx3
.
0.0
-
dtk22
.
10.1
...
@@ -28,6 +28,13 @@ docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --devi
...
@@ -28,6 +28,13 @@ docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --devi
# 激活dtk
# 激活dtk
source /opt/dtk/env.sh
source /opt/dtk/env.sh
```
```
### Dockerfile(方法二)
```
cd ./docker
docker build --no-cache -t resnet50_onnxruntime:2.0 .
docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/resnet50_onnxruntime:/home/resnet50_onnxruntime -it <Your Image ID> /bin/bash
```
## 数据集
## 数据集
Imagenet-1k
Imagenet-1k
...
@@ -102,6 +109,8 @@ class=n01843065 jacamar ; probability=11.903831
...
@@ -102,6 +109,8 @@ class=n01843065 jacamar ; probability=11.903831
```
```
class=n01843065 jacamar ; probability=11.903831
class=n01843065 jacamar ; probability=11.903831
```
```
### 精度
无
## 应用场景
## 应用场景
### 算法类别
### 算法类别
...
...
docker/Dockerfile
0 → 100644
View file @
17fa8aed
FROM
image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
RUN
source
/opt/dtk/env.sh
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