yolov3-channel-and-layer-pruning icon indicating copy to clipboard operation
yolov3-channel-and-layer-pruning copied to clipboard

Published 20 hours ago • verified publisher icon tanluren

想请教一下onnx剪枝问题

Open HenryWang628 opened this issue 4 years ago • 2 comments

想请教一下如何把剪枝以后的模型保存为onnx呀，是在剪枝进行之前就必须保存为onnx还是pt文件剪完枝后再转onnx呢

HenryWang628 avatar Jul 05 '20 14:07 HenryWang628

建议剪完之后再转onnx。这个框架似乎不支持onnx剪枝。

zhuhuangru avatar Jul 06 '20 09:07 zhuhuangru

建议剪完之后再转onnx。这个框架似乎不支持onnx剪枝。

好的十分感谢，那我就用稀疏训练完的pt文件——>剪枝——>再转onnx就可以了。 还想请教一下剪完枝的模型是可以直接用ultralytics的model.py转onnx的吗

HenryWang628 avatar Jul 06 '20 10:07 HenryWang628