nnieqat-pytorch icon indicating copy to clipboard operation
nnieqat-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon aovoc

量化感知训练后模型的部署

Open Timothy-105 opened this issue 3 years ago • 3 comments

量化感知训练后将模型部署在hisi平台，pytorch 转成caffe的过程中，相比原来的模型，在卷积激活后会有一个cast层，caffe不支持，请问可有解决的办法？

Timothy-105 avatar Jul 15 '21 13:07 Timothy-105

请问解决了吗

forwardwfg avatar Sep 18 '21 01:09 forwardwfg

我走的是pytorch->caffe 的流程，没碰到这个问题哈，具体解决需要看一下这个cast 层怎么产生，功能是什么

aovoc avatar Sep 20 '21 13:09 aovoc

量化感知训练后将模型部署在hisi平台，pytorch 转成caffe的过程中，相比原来的模型，在卷积激活后会有一个cast层，caffe不支持，请问可有解决的办法？

你有使用onnx-simplifier吗

jie311 avatar Sep 23 '21 09:09 jie311