请教一下基于pytorch框架的模型迁移问题。
收藏回复举报
请教一下基于pytorch框架的模型迁移问题。
t('forum.solved') 已解决
新人帖
发表于2024-07-10 21:25:24
0 查看

任务目标:

将原有pytorch框架下的模型与算法迁移到国产昇腾环境运行。

任务进度:

参考官方文档模型转换教程,已经成功转为om格式模型,也用简单图片分类例子测试过。

任务问题:

①torch_npu是否能直接支持原有框架,可以最大化避免代码改动?

②用的简单测试例子只有一张图片输入,看代码使用了acl库,有初始化、释放等操作,该方式报错时候很难定位具体原因,输入输出似乎有格式要求,但我没发现那种明确的transform函数。想知道是否有其他更方便的工具,以及是否有包含模型转换、代码转换、数据预处理、分批加载等步骤的更完整的参考样例。

③原算法中包含JSMA对抗攻击算法需要计算模型梯度,也要用到模型输出,这部分代码转换有无专门的工具包?

我要发帖子