@Amrit_Das Seems like the following line is the culprit ? How should I change this ?

  
      gitlab.com
  
  
    promach/Pruning-CNN/blob/master/SqueezeNet-Pruning/prune.py#L180 1


            [new_conv_layer]) for i, _ in enumerate(model.classifier)))


    del model.classifier
    del next_conv
    del conv
    model.classifier = classifier


return model


if __name__ == '__main__':
model = models.vgg16(pretrained=True)
model.train()


t0 = time.time()
model = prune_conv_layer(model, 28, 10)
print("The prunning took %d\n", time.time() - t0)