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_tensorflow
Commits
4ae16240
Commit
4ae16240
authored
Aug 25, 2023
by
“qianyj”
Browse files
Update README
parent
5ab49e19
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
4 deletions
+5
-4
README.md
README.md
+5
-4
No files found.
README.md
View file @
4ae16240
...
...
@@ -17,7 +17,7 @@ ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.7.0-centos7.6-dtk-22.10.1-py38-latest
# <Your Image ID>用上面拉取docker镜像的ID替换
docker run --shm-size 16g --network=host --name=resnet50_tensorFlow --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/resnet50_tensorflow:/home/resnet50_tensorflow -it <Your Image ID> bash
pip install -r requirements.txt
pip install -r requirements.txt
--no-deps
```
### Dockerfile(方法二)
```
...
...
@@ -27,12 +27,12 @@ docker run --rm --shm-size 16g --network=host --name=resnet50_tensorflow --privi
```
### Anaconda(方法三)
1、关于本项目DCU显卡所需的特殊深度学习库可以从开发
社
社区下载安装:
1、关于本项目DCU显卡所需的特殊深度学习库可以从开发
者
社区下载安装:
https://developer.hpccube.com/tool/
```
DTK版本:dtk22.10.1
python: 3.8
tensorflow: 2.
9
tensorflow: 2.
7
tf-models-official: 2.7
keras: 2.7
tensorboard: 2.7
...
...
@@ -50,6 +50,7 @@ pip3 install -r requirements.txt --no-deps
ImageNet数据集可以
[
官网
](
https://image-net.org/
"ImageNet数据集官网"
)
下载、百度搜索或者联系我们
ImageNet数据集转成TFRecord格式,可以参考以下
[
script
](
https://github.com/tensorflow/tpu/blob/master/tools/datasets/imagenet_to_gcs.py
)
和
[
README
](
https://github.com/tensorflow/tpu/tree/master/tools/datasets#imagenet_to_gcspy
)
制作完成的TFRrecord数据形式如下:
```
tfrecord-imagenet
|
train-00000-of-01024
...
...
@@ -60,7 +61,7 @@ tfrecord-imagenet
validation-00001-of-00128
...
validation-00127-of-00128
```
2、合成数据
基于随机合成的数据,不需要下载ImageNet数据集,执行网络训练时只需要把程序执行语句中的--use_synthetic_data设置为true即可
...
...
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