【2023 · CANN训练营第一季】昇腾AI入门课(PyTorch)第一章结课作业学习笔记



遇到的问题是:第一次用atc转化模型显示不能用,使用source ~/bashrc调了一下环境就好了
优化:在类的构造函数和析构函数中添加了注释,解释了它们的目的。
在InitResource函数中添加了注释,描述了初始化过程的每个步骤。
在Process函数中添加了注释,解释了模型处理的步骤。
在DestroyResource函数中添加了注释,描述了释放分配的资源的过程。
在需要释放资源的地方添加对应的释放操作,防止内存泄漏。
使用日志信息标记初始化、销毁和处理过程的关键步骤,方便调试和追踪代码执行流程。
提示更好的代码组织结构,将代码按逻辑分块,并在主函数中调用这些逻辑块,使代码更易于阅读和维护。
对函数和类添加必要的文档注释,描述其作用和用法,提高代码的可读性和可维护性。对ACL函数调用进行了错误处理,并在必要时记录错误消息
2
转换命令:atc --model=caffe_model/resnet101_tf.pb --framework=3 --output=model/resnet101_tf --output_type=FP32 --soc_version=Ascend310 --input_shape="input:1,224,224,3" --log=info
转化成功截图:

编译运行成功截图:

3.

4.
#·进入脚本转换工具所在路径
cd /home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/ms_fmk_transplt
# 执行脚本转换。
./pytorch_gpu2npu.sh -i /home/HwHiAiUser/LeNet -o /home/HwHiAiUser/LeNet_output

采用加一句话迁移的方式,在脚本中添加:
from torch_npu.contrib import transfer_to_npu

5.
#·进入脚本转换工具所在路径
cd /home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/ms_fmk_transplt
# 执行脚本转换。
./pytorch_gpu2npu.sh -i /home/HwHiAiUser/ddp -o /home/HwHiAiUser/ddp_output

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。
更多推荐


所有评论(0)