文件最后提交记录最后更新时间
!6851 [Clean Code] 修改芯片型号描述 Merge pull request !6851 from 施康/master 1 年前
!2082 ACL_PyTorch推理模型环境变量批量整改 * ACL_PyTorch推理模型环境变量整改 3 年前
init 4 年前
!2082 ACL_PyTorch推理模型环境变量批量整改 * ACL_PyTorch推理模型环境变量整改 3 年前
README.md

环境准备:

1.数据集路径
数据集统一放在/root/datasets/或/opt/npu/
本模型数据集放在/root/datasets/

2.进入工作目录
cd ResNet18

3.安装必要的依赖
pip3.7 install -r requirements.txt

4.获取模型代码
git clone https://github.com/pytorch/vision

5.如果模型代码需要安装,则安装模型代码
cd vision
python3.7 setup.py install
cd ..

6.获取权重文件
wget https://download.pytorch.org/models/resnet18-f37072fd.pth

7.获取benchmark工具
将benchmark.x86_64放在当前目录

8.执行时确保device空闲
bash test/pth2om.sh
bash test/eval_acc_perf.sh --datasets_path=/root/datasets