在此一键设置昵称,即可参与社区互动!
AI_学习者
发帖: 11粉丝: 1
发消息 + 关注
请问下,这个TransData是什么算子,将近占了训练总耗时的50%,我代码里并没有使用这个,文档上也查不到这个算子的信息
看到好多地方,BN算子等或者一些卷积,都调用了这个
有么有啥办法优化一下
分享文章到朋友圈
chengxiaoli
发帖: 154粉丝: 25
级别 : 版主,版块专家
您好,欢迎您的提问。
正在定位问题原因,会尽快给您答复。
白鹿第一帅
发帖: 1粉丝: 5
Mark
梁成辉
发帖: 0粉丝: 1
transdata算子出现的场景是:如果网络中相互连接的算子使用的数据格式不一致(如NC1HWC0),框架就会自动插入transdata算子使其转换成一致的数据格式,然后再进行计算。
可以考虑训练的时候用我们的amp做混合精度,这样能减少一些fp32的运算,应该能减少一些transdata算子的调用。
游客
回复主题
您发表的内容存在敏感词汇
如点击继续发布,文章中的敏感词将以*代替。