剪枝遇到问题:
RuntimeError: Given groups=1, weight of size [64, 64, 3, 3], expected input[1, 28, 80, 80] to have 64 channels, but got 28 channels instead
错误如上所述。
解决办法:
这里的threshold×0.8即可。
其他系数没试过,你也可以试试。
分析:
只是个人理解:
随着factor的提升,input channels也增高,说明剪枝剪多了
随着threshold的降低,input channels降低。