Commit 6bba0afd authored by “yuguo”'s avatar “yuguo”
Browse files

update

parent b80c5d6f
...@@ -27,6 +27,29 @@ cd /home/resnet50_oneflow ...@@ -27,6 +27,29 @@ cd /home/resnet50_oneflow
pip install -r requirements.txt pip install -r requirements.txt
``` ```
### Conda
1. 创建conda虚拟环境:
```plaintext
conda create -n resnet python=3.9
```
关于本项目DCU显卡所需的工具包、深度学习库等均可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
- [DTK-22.10.1](https://cancon.hpccube.com:65024/1/main/DTK-22.10.1)
- [Oneflow-0.9](https://cancon.hpccube.com:65024/4/main/oneflow/dtk22.10)
Tips:以上dtk驱动、python等工具版本需要严格一一对应。
其它依赖库参照requirements.txt安装:
```plaintext
cd resnet50_oneflow
pip install -r requirements.txt
```
## 数据集 ## 数据集
我们考虑到imagenet数据集比较庞大,为了用户可以使用OneFlow快速进行Resnet50的训练验证,采用mini-imagenet小数据集。如果需要原始数据需要参考该链接https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnet50#prepare-ofrecord-for-the-full-imagenet-dataset进行转换为ofrecord 我们考虑到imagenet数据集比较庞大,为了用户可以使用OneFlow快速进行Resnet50的训练验证,采用mini-imagenet小数据集。如果需要原始数据需要参考该链接https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnet50#prepare-ofrecord-for-the-full-imagenet-dataset进行转换为ofrecord
...@@ -99,7 +122,7 @@ ofrecord-imagenet ...@@ -99,7 +122,7 @@ ofrecord-imagenet
- https://developer.hpccube.com/codes/modelzoo/resnet50_oneflow - https://developer.hpccube.com/codes/modelzoo/resnet50_oneflow
## 参考 ## 参考资料
* https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnet50 * https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnet50
* https://github.com/Oneflow-Inc/oneflow * https://github.com/Oneflow-Inc/oneflow
\ No newline at end of file
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