Does channel prune rmt support nchw model?

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

Does channel prune rmt support nchw model? #302

Open ChaoYu123 opened 5 years ago

ChaoYu123 commented 5 years ago

I use a NCHW model as input for channel_pruning_rmt and change the corresponding index of tensors. However, I fail at "assert err_full < 1e-6, 'unable to recover output feature maps - full (%e)' % err_full" in __smpl_inputs_n_outputs function. I wonder if channel pruning rmt supports NCHW model for channel pruning.