search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

想请教一下onnx剪枝问题 #92

Open HenryWang628 opened 4 years ago

HenryWang628 commented 4 years ago

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

zhuhuangru commented 4 years ago

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

HenryWang628 commented 4 years ago

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

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