Exception while running channel pruning using Pytorch

@quic_sundarr

I have shared the model files in another thread here: Exception while exporting Quantized model - #5 by amitdedhia

Can you pick from it? - Thanks

I will try just Channel Pruning later (soon) and let you know the results