@Amrit_Das @albanD
See this diffcheck result 105 between original SqueezeNet model (left)  and pruned SqueezeNet model (right)
So, how do I go around this restriction (tensor size difference) in actual python coding (how to modify the model loading code segment in predict.py 29 for the smaller, pruned SqueezeNet model) ?