Amrit_Das:

Rather it would be advisable to choose similar kind of model.


@Amrit_Das  what do you exactly mean by similar model ? I am doing SqueezeNet model pruning here, therefore there is not going to be any existing model that will fit my model_prunned 100% without any tensor size mismatch.
This is the problem: How am I going to load my model_prunned ? and one more problem, the pruning is not deterministic, therefore, manual crafting of the tensor size to fit the model_prunned will not be possible for every pruning process.