logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

使用ModelArt平台昇腾芯片,将Restnet50模型迁移到猫狗分类实验

总体实验较为简单,数据集处理较为繁琐,在书写Dataset和DataLoader时需要不断考虑NPU是否可以适配,后经过torch_npu的自动迁移,发现无需将cuda相关的函数以及字符修改,自动迁移可以进行自动更换,方便我们使用。其次需要将模型迁移到不同数据集时更换预训练的网络层,以避免输出与label不匹配。Ascend使用整体情况,发现模型在npu’和cpu之间转换时耗时太长,同样的实验环境

文章图片
#分类#数据挖掘#人工智能
到底了