任务目标:
将原有pytorch框架下的模型与算法迁移到国产昇腾环境运行。
任务进度:
参考官方文档模型转换教程,已经成功转为om格式模型,也用简单图片分类例子测试过。
任务问题:
①torch_npu是否能直接支持原有框架,可以最大化避免代码改动?
②用的简单测试例子只有一张图片输入,看代码使用了acl库,有初始化、释放等操作,该方式报错时候很难定位具体原因,输入输出似乎有格式要求,但我没发现那种明确的transform函数。想知道是否有其他更方便的工具,以及是否有包含模型转换、代码转换、数据预处理、分批加载等步骤的更完整的参考样例。
③原算法中包含JSMA对抗攻击算法需要计算模型梯度,也要用到模型输出,这部分代码转换有无专门的工具包?
任务目标:
将原有pytorch框架下的模型与算法迁移到国产昇腾环境运行。
任务进度:
参考官方文档模型转换教程,已经成功转为om格式模型,也用简单图片分类例子测试过。
任务问题:
①torch_npu是否能直接支持原有框架,可以最大化避免代码改动?
②用的简单测试例子只有一张图片输入,看代码使用了acl库,有初始化、释放等操作,该方式报错时候很难定位具体原因,输入输出似乎有格式要求,但我没发现那种明确的transform函数。想知道是否有其他更方便的工具,以及是否有包含模型转换、代码转换、数据预处理、分批加载等步骤的更完整的参考样例。
③原算法中包含JSMA对抗攻击算法需要计算模型梯度,也要用到模型输出,这部分代码转换有无专门的工具包?