search

Tencent / PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.
https://pocketflow.github.io
Other
2.79k stars 490 forks source link

cifar10_channel pruned 的示例，通道剪枝(channel_pruning) 导出修改了计算图之后，速度比之前的更慢了！ #307

Closed hudengjunai closed 5 years ago

hudengjunai commented 5 years ago

如题所示：对cifar10-resnet18 例子中模型进行通道剪枝，由于剪枝之后模型更改中将全为0的filter和对应feature_map通道删除，采用了1x1 kernel filter channel 选择方式，相比之前增加了一层。导出的frozen pb 计算图在GPU上速度变慢了，有什么解决方案吗？

hudengjunai commented 5 years ago

如题所示：对cifar10-resnet18 例子中模型进行通道剪枝，由于剪枝之后模型更改中将全为0的filter和对应feature_map通道删除，采用了1x1 kernel filter channel 选择方式，相比之前增加了一层。导出的frozen pb 计算图在GPU上速度变慢了，有什么解决方案吗？ 问题已经在 #119 解决了。